Function count_parameters

utils/utils.py:44–45 · view source on GitHub ↗

(model)

Source from the content-addressed store, hash-verified

42
43
44	def count_parameters(model):
45	return sum(p.numel() for p in model.parameters() if p.requires_grad)

fitFunction · 0.90

no outgoing calls

no test coverage detected