Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.checkpoint.UploadTaskGroupBuilder Class Reference
Inheritance diagram for caffe2.python.checkpoint.UploadTaskGroupBuilder:

Public Member Functions

def build (self, epoch, checkpoint_manager)
 

Detailed Description

A simple class to upload checkpoints.

Definition at line 561 of file checkpoint.py.

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.

Definition at line 563 of file checkpoint.py.


The documentation for this class was generated from the following file: