From 9b1792fcbe9737d356e31b4aec574bad7c58fd58 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 27 Aug 2024 17:01:13 +0200 Subject: [PATCH] Update. --- main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/main.py b/main.py index 0ecd492..e4237d2 100755 --- a/main.py +++ b/main.py @@ -941,6 +941,7 @@ def ae_generate(model, input, mask_generate, noise_proba, nb_iterations_max=50): update = (1 - mask_to_change) * input + mask_to_change * final if update.equal(input): + log_string(f"exit after {it+1} iterations") break else: changed = changed & (update != input).max(dim=1).values -- 2.39.5