oneflow.train

class oneflow.train.CheckPoint

Create a CheckPoint object to manage checkpoint manually.

__init__() → None

Initialize self. See help(type(self)) for accurate signature.

init() → None

Initialize models by default initializer of op or Job.

load(path: str) → None

load a checkpoint from path and initialize models.

Parameters

path – A string of path to load checkpoint.

save(path: str) → None

save a checkpoint to path.

Parameters

path – A string of path to save checkpoint.

class oneflow.train.SimpleCheckPointManager(root_path: str, prefix: str = 'snapshot_')

SimpleCheckPointManager is a simple automatic checkpoint manager.

Parameters

• root_path – root path of snapshot

• prefix – prefix of snapshot

__init__(root_path: str, prefix: str = 'snapshot_') → None

Initialize self. See help(type(self)) for accurate signature.

initialize_or_restore() → None

latest_checkpoint() → Optional[str]

list_checkpoints() → List[str]

save() → None