Method state_dict

DeepSpeedExample/megatron/model/module.py:181–182

(self, destination=None, prefix='', keep_vars=False)

    def state_dict(self, destination=None, prefix='', keep_vars=False):
        return self.module.state_dict(destination, prefix, keep_vars)


    def state_dict_for_save_checkpoint(self, destination=None, prefix='',