Method state_dict

codegeex/megatron/optimizer/optimizer.py:501–502 · view source on GitHub ↗

(self)

Source from the content-addressed store, hash-verified

499	pass
500
501	def state_dict(self):
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

state_dictMethod · 0.45

no outgoing calls

no test coverage detected