Public Member Functions

Detailed Description

Member Function Documentation

def caffe2.python.checkpoint.UploadTaskGroupBuilder.build

(

self,

epoch,

checkpoint_manager

)

Builds the task group to upload checkpoints.
Args:
epoch: An integer. The checkpoint epoch to be uploaded.
checkpoint_manager: Can be a CheckpointManager for single machine
or a MultiNodeCheckpointManager for multi-machine. The manager
that initializes/saves/loads checkpoints.
Raises:
NotImplementedError: This base class only has the interface,
the implementation will be in the subclasses.