Refactor n_parameters of TorchComposableInfluence to be callable befo…

…re fit
aai-institute · Jun 13, 2024 · ef32847 · ef32847
1 parent 6e20ec5
commit ef32847
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/pydvl/influence/torch/base.py b/src/pydvl/influence/torch/base.py
@@ -724,7 +724,11 @@ def block_names(self) -> List[str]:
 
     @property
     def n_parameters(self):
-        return sum(block.op.input_size for _, block in self.block_mapper.items())
+        return sum(
+            param.numel()
+            for block in self.parameter_dict.values()
+            for param in block.values()
+        )
 
     @abstractmethod
     def with_regularization(