2 lines
101 B
Python
2 lines
101 B
Python
def count_parameters(model):
|
|
return sum(p.numel() for p in model.parameters() if p.requires_grad) |