CheckpointSchedule

class CheckpointSchedule[source]

Bases: ABC

Interface for checkpoint schedules.

Methods Summary

__call__(step)

Decide whether to create a checkpoint at the specified epoch.

Methods Documentation

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

Decide whether to create a checkpoint at the specified epoch.

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