Update.

diff --git a/main.py b/main.py
index b9ec213..27817e8 100755 (executable)
--- 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