ModelCheckpoint¶

class cl_gym.utils.callbacks.ModelCheckpoint(interval='task', name_prefix=None)[source]¶

Bases: cl_gym.utils.callbacks.base.ContinualCallback

Model Checkpoint callback. Saves the backbone (model) to disk at the end of every epoch or task (depending on the input settings).