Method state_dict

TinyViT/utils.py:323–324 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

321	return norm
322
323	def state_dict(self):
324	return self._scaler.state_dict()
325
326	def load_state_dict(self, state_dict):
327	self._scaler.load_state_dict(state_dict)

load_checkpointFunction · 0.45

load_pretrainedFunction · 0.45

save_checkpointFunction · 0.45

no outgoing calls

no test coverage detected