From: François Fleuret 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