From acec60ad56f75cff89f94b7957ba35b244132e44 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Tue, 24 Sep 2024 12:20:43 +0200 Subject: [PATCH] Update. --- main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/main.py b/main.py index 09e3c59..b2b0492 100755 --- a/main.py +++ b/main.py @@ -877,7 +877,7 @@ def inject_plasticity(model, proba): with torch.no_grad(): for p, q in zip(model.parameters(), dummy.parameters()): mask = (torch.rand(p.size()) <= proba).long() - p[...] = (1 - mask) * p + mmask * q + p[...] = (1 - mask) * p + mask * q ###################################################################### -- 2.39.5