BestCheckpointSchedule

class BestCheckpointSchedule(result_tracker: ResultTracker, metric_selection: MetricSelection)

Bases: CheckpointSchedule

Create a checkpoint whenever a metric improves.

Methods Summary

__call__(step)

Decide whether to create a checkpoint at the specified epoch.

Methods Documentation

Parameters:

• result_tracker (ResultTracker)

• metric_selection (MetricSelection)

__call__(step: int) → bool

Decide whether to create a checkpoint at the specified epoch.

Parameters:

step (int)

Return type:

bool