From: François Fleuret <francois@fleuret.org>
Date: Thu, 5 Sep 2024 12:48:44 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=57d8f6716cff814dc520472967d64a0c8a8fc58d;p=culture.git

Update.
---

diff --git a/main.py b/main.py
index 8562e32..174b9b8 100755
--- a/main.py
+++ b/main.py
@@ -1429,6 +1429,9 @@ for n_epoch in range(current_epoch, args.nb_epochs):
 
     threads = []
 
+    # for model in models:
+    # log_string(f"DEBUG {model.id} {sum([ p.sum() for p in model.parameters()]).item()}")
+
     start_time = time.perf_counter()
 
     for gpu, model in zip(gpus, weakest_models):