mypy

Lightning-AI · Jun 23, 2022 · 7eba6a4 · 7eba6a4
1 parent 5b5a58e
commit 7eba6a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pytorch_lightning/utilities/model_summary.py b/src/pytorch_lightning/utilities/model_summary.py
@@ -120,7 +120,7 @@ def layer_type(self) -> str:
     @property
     def num_parameters(self) -> int:
         """Returns the number of parameters in this module."""
-        return sum(np.prod(p.shape) if not _is_lazy_weight_tensor(p) else 0 for p in self._module.parameters())
+        return int(sum(np.prod(p.shape) if not _is_lazy_weight_tensor(p) else 0 for p in self._module.parameters()))
 
 
 class ModelSummary: