Method load_state_dict

star/utils/misc.py:276–277 · view source on GitHub ↗

(self, state_dict)

Source from the content-addressed store, hash-verified

274	return self._scaler.state_dict()
275
276	def load_state_dict(self, state_dict):
277	self._scaler.load_state_dict(state_dict)
278
279
280	def get_grad_norm_(parameters, norm_type: float = 2.0) -> torch.Tensor:

mainFunction · 0.80

load_modelFunction · 0.80

resume_from_cpuMethod · 0.80

resumeMethod · 0.80

no outgoing calls

mainFunction · 0.64