MCPcopy Create free account
hub / github.com/AIS-SNU/Smart-Infinity / state_dict

Method state_dict

DeepSpeedExample/megatron/fp16/fp16.py:76–77  ·  view source on GitHub ↗
(self, destination=None, prefix='', keep_vars=False)

Source from the content-addressed store, hash-verified

74 return fp16_to_fp32(self.module(*(fp32_to_fp16(inputs)), **kwargs))
75
76 def state_dict(self, destination=None, prefix='', keep_vars=False):
77 return self.module.state_dict(destination, prefix, keep_vars)
78
79 def state_dict_for_save_checkpoint(self, destination=None, prefix='',
80 keep_vars=False):

Callers 4

save_checkpointFunction · 0.45
__init__Method · 0.45
state_dictMethod · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected