trainers.DifftreParallel.checkpoint

DifftreParallel.checkpoint(name, object)

Marks attribute to be saved in a partial checkpoint.

The marked attribute is saved to a checkpoint dictionary under the specified name.

Parameters:

• name – Name of the statistic in the saved dictionary

• object – Object to initialize the attribute

Returns:

Returns the original object wrapped as a CheckpointNode.