Function count_parameters

tools/latency.py:34–35 · view source on GitHub ↗

(model)

Source from the content-addressed store, hash-verified

32
33
34	def count_parameters(model):
35	return sum(p.numel() for p in model.parameters() if p.requires_grad)
36
37
38	def main():

mainFunction · 0.85

no outgoing calls

no test coverage detected