MCPcopy Create free account
hub / github.com/MotrixLab/ViMoGen / state_dict

Method state_dict

datasets/sampler.py:355–356  ·  view source on GitHub ↗
(self, step)

Source from the content-addressed store, hash-verified

353 self.start_index = 0
354
355 def state_dict(self, step) -> dict:
356 return {'start_index': step}
357
358 def load_state_dict(self, state_dict: dict):
359 self.start_index = state_dict['start_index'] + 1

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected