From: François Fleuret <francois@fleuret.org>
Date: Tue, 24 Sep 2024 06:18:41 +0000 (+0200)
Subject: Update.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=377f3de44ce792ec9b92ca3d2e9be93a33e3ddac;p=culture.git

Update.
---

diff --git a/main.py b/main.py
index 3424b27..09e3c59 100755
--- a/main.py
+++ b/main.py
@@ -875,7 +875,7 @@ def inject_plasticity(model, proba):
     dummy = new_model()
 
     with torch.no_grad():
-        for p, q in zip(mode.parameters(), dummy.parameters()):
+        for p, q in zip(model.parameters(), dummy.parameters()):
             mask = (torch.rand(p.size()) <= proba).long()
             p[...] = (1 - mask) * p + mmask * q
 
@@ -907,8 +907,7 @@ models = []
 
 for i in range(args.nb_models):
     model = new_model(i)
-    model = torch.compile(model)
-
+    # model = torch.compile(model)
     models.append(model)
 
 ######################################################################