Function count_params

diff2flow/helpers.py:114–115 · view source on GitHub ↗

(model)

Source from the content-addressed store, hash-verified

112
113
114	def count_params(model):
115	return sum(p.numel() for p in model.parameters() if p.requires_grad)
116
117
118	def timer(start, end):

mainFunction · 0.90

no outgoing calls

no test coverage detected