Function cnt_params

utils/train_state_utils.py:67–68 · view source on GitHub ↗

(model)

Source from the content-addressed store, hash-verified

65
66
67	def cnt_params(model):
68	return sum(param.numel() for param in model.parameters())
69
70
71	def initialize_train_state(config, model, model_ema , device):

initialize_train_stateFunction · 0.85

no outgoing calls

no test coverage detected