Method load_state_dict

codegeex/megatron/optimizer/optimizer.py:504–505 · view source on GitHub ↗

(self, state_dict)

Source from the content-addressed store, hash-verified

502	return self.optimizer.state_dict()
503
504	def load_state_dict(self, state_dict):
505	self.optimizer.load_state_dict(state_dict)

__init__Method · 0.45

load_state_dictMethod · 0.45

no outgoing calls

no test coverage detected