EveryCheckpointSchedule

class EveryCheckpointSchedule(frequency: int = 10)[source]

Bases: CheckpointSchedule

Create a checkpoint every \(n\) steps.

Attributes Summary

frequency

The checkpoint frequency

Methods Summary

__call__(step)

Decide whether to create a checkpoint at the specified epoch.

Attributes Documentation

Parameters:

frequency (int)

frequency: int = 10

The checkpoint frequency

Methods Documentation

__call__(step: int) → bool[source]

Decide whether to create a checkpoint at the specified epoch.

Parameters:

step (int)

Return type:

bool