From b93eec2e3cd0135486e9fabfe44a37f89d2ee64a Mon Sep 17 00:00:00 2001
From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= <francois@fleuret.org>
Date: Mon, 26 Aug 2024 14:57:52 +0200
Subject: [PATCH] Update.

---
 main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/main.py b/main.py
index b9ec213..27817e8 100755
--- a/main.py
+++ b/main.py
@@ -915,6 +915,7 @@ def ae_generate(
 
         update = (1 - mask_to_change) * input + mask_to_change * dist.sample()
         if update.equal(input):
+            log_string(f"converged at iteration {it}")
             break
         else:
             changed = changed & (update != input).max(dim=1).values
-- 
2.39.5