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 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 639 of file checkpoint.py.

---

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

• caffe2/python/checkpoint.py