UnionCheckpointSchedule

class UnionCheckpointSchedule(bases: str | CheckpointSchedule | type[CheckpointSchedule] | None | Sequence[str | CheckpointSchedule | type[CheckpointSchedule] | None], bases_kwargs: Mapping[str, Any] | None | Sequence[Mapping[str, Any] | None] = None)

Bases: CheckpointSchedule

Create a checkpoint whenever one of the base schedules requires it.

Attributes Summary

bases_kwargs

Methods Summary

__call__(step)

Decide whether to create a checkpoint at the specified epoch.

Attributes Documentation

Parameters:

• bases (str | CheckpointSchedule | type[CheckpointSchedule] | None | Sequence[str | CheckpointSchedule | type[CheckpointSchedule] | None])

• bases_kwargs (Mapping[str, Any] | None | Sequence[Mapping[str, Any] | None])

bases_kwargs: Mapping[str, Any] | None | Sequence[Mapping[str, Any] | None] = None

Methods Documentation

__call__(step: int) → bool

Decide whether to create a checkpoint at the specified epoch.

Parameters:

step (int)

Return type:

bool