clean up

2026-07-08 20:36:49 +02:00
parent 90e33f074d
commit 31f461342e
3 changed files with 70 additions and 147 deletions
--- a/unlearning/LinearFiltration.py
+++ b/unlearning/LinearFiltration.py
@@ -12,10 +12,6 @@ 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
-
        device = next(model.parameters()).device
            
        return self.normalise(