ExplicitCheckpointSchedule

class ExplicitCheckpointSchedule(steps: Collection[int])[source]

Bases: CheckpointSchedule

Create a checkpoint for explicitly chosen steps.

Methods Summary

__call__(step)

Decide whether to create a checkpoint at the specified epoch.

Methods Documentation

Parameters:: steps (Collection[int])

__call__(step: int) → bool[source]

Decide whether to create a checkpoint at the specified epoch.

Parameters:: step (int)
Return type:: bool