attck metrics

unlearning/LinearFiltration.py

@@ -13,8 +13,8 @@ class LinearFiltration(Strategy):
     def _run(self, model: nn.Module, forget_loader: DataLoader, retain_loader: DataLoader) -> nn.Module:
         model.eval()
         # Freeze internal params
-        for param in model.parameters():
-            param.requires_grad = False
+        #for param in model.parameters():
+            #param.requires_grad = False
 
         device = next(model.parameters()).device
             
@@ -155,7 +155,8 @@ class LinearFiltration(Strategy):
 
         # 12
         clf = self._get_classifier(model)
-        clf.weight.copy_(W_Z)
+        with torch.no_grad():
+            clf.weight.copy_(W_Z)
 
         return model