From: François Fleuret <francois@fleuret.org>
Date: Mon, 26 Aug 2024 12:57:52 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=b93eec2e3cd0135486e9fabfe44a37f89d2ee64a;p=culture.git

Update.
---

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