ksuit.callbacks.checkpoint_callbacks.checkpoint

Classes

CheckpointCallback

Callback to save the model and optimizer state periodically.

Module Contents

class ksuit.callbacks.checkpoint_callbacks.checkpoint.CheckpointCallback(callback_config, **kwargs)

Bases: ksuit.callbacks.base.periodic_callback.PeriodicCallback

Callback to save the model and optimizer state periodically.

Initializes the CheckpointCallback.

Parameters:

• callback_config (ksuit.schemas.callbacks.callbacks_config.CheckpointCallbackConfig) – The configuration for the callback. Implements the CheckpointCallbackConfig schema.

• **kwargs – additional arguments passed to the parent class.

save_weights

save_optim

save_latest_weights

save_latest_optim

model_names = []