From 377f3de44ce792ec9b92ca3d2e9be93a33e3ddac Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 24 Sep 2024 08:18:41 +0200 Subject: [PATCH] Update. --- main.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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) ###################################################################### -- 2.39.5