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 637 of file

Member Function Documentation

def (   self,
Builds the task group to upload checkpoints.

    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.

    NotImplementedError: This base class only has the interface,
the implementation will be in the subclasses.

Definition at line 639 of file

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