core.trainers.mixins.checkpoints

core.trainers.mixins.checkpoints

Custom handling to not fail training if fsdp optimizer is not savable

Classes

Name	Description
CheckpointSaveMixin	Mixin to handle saving the optimizer and scheduler if they are not savable.

CheckpointSaveMixin

core.trainers.mixins.checkpoints.CheckpointSaveMixin()

Mixin to handle saving the optimizer and scheduler if they are not savable.