Skip to content

Commit bdce2cb

Browse files
committed
Add get_num_parameters
1 parent 88fe146 commit bdce2cb

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

squadro/evaluators/evaluator.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -116,3 +116,7 @@ def save(self, filepath: str | Path, weights_only=False):
116116

117117
def byte_size(self, human_readable=True) -> int | str:
118118
return get_model_size(self, human_readable)
119+
120+
121+
def get_num_parameters(model):
122+
return sum(p.numel() for p in model.parameters())

0 commit comments

Comments
 (0)