Update.

diff --git a/main.py b/main.py
index 3424b27..09e3c59 100755 (executable)
--- 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)
 
 ######################################################################