pytorch_pfn_extras.training.triggers.ManualScheduleTrigger

class pytorch_pfn_extras.training.triggers.ManualScheduleTrigger(points, unit)

Trigger invoked at specified point(s) of iterations or epochs.

This trigger accepts iterations or epochs indicated by given point(s). There are two ways to specify the point(s): iteration and epoch. iteration means the number of updates, while epoch means the number of sweeps over the training dataset. Fractional values are allowed if the point is a number of epochs; the trigger uses the iteration and epoch_detail attributes defined by the manager.

Parameters

• points (int, float, or list of int or float) – time of the trigger. Must be an integer or list of integer if unit is 'iteration'.

• unit (str) – Unit of the time specified by points. It must be either 'iteration' or 'epoch'.

finished

Flag that indicates whether or not this trigger will

Type

bool

fire in the future. This flag is used to determine if the extension

should be initialized after resume.

__init__(points, unit)

Methods

__init__(points, unit)
load_state_dict(to_load)
state_dict()

pytorch_pfn_extras.training.triggers.IntervalTrigger pytorch_pfn_extras.training.triggers.BestValueTrigger