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'.

__init__(points, unit)
Parameters

  • points (Union[float, Sequence[float]]) –

  • unit (UnitLiteral) –

Methods

__init__(points, unit)

load_state_dict(state)

may_fire(iteration, epoch_length)

Flags if the trigger may fire at the current iteration

state_dict()

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