MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / load

Method load

python/oneflow/framework/model.py:560–563  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

558 super().__init__("checkpointing", cfg, model, callbacks)
559
560 def load(self):
561 assert self.is_valid
562 if self._cfg.need_load:
563 self._load_checkpoint(self._cfg.load_dirpath)
564
565 def step(self, step_idx: int = 0):
566 assert self.is_valid

Callers 6

is_oneflow_pickle_fileFunction · 0.45
fitMethod · 0.45
_load_checkpointMethod · 0.45
_load_checkpointMethod · 0.45
warmup_with_loadMethod · 0.45

Calls 1

_load_checkpointMethod · 0.95

Tested by

no test coverage detected