Function count_model_parameters

train_eval_vimogen.py:36–38 · view source on GitHub ↗

(model: nn.Module)

Source from the content-addressed store, hash-verified

34
35
36	def count_model_parameters(model: nn.Module):
37	params = sum([np.prod(p.size()) for p in model.parameters()])
38	return params
39
40
41	def sample_data(loader, sampler, start_epoch, start_iter):

mainFunction · 0.85

no outgoing calls

no test coverage detected