|View source on GitHub|
Provides TFM orbit actions and associated helper functions/classes.
class EMACheckpointing: Eval action to save checkpoint with average weights when EMA is used.
class PruningAction: Train action to updates pruning related information.
class RecoveryAction: Train action to recover from loss blowup.
class RecoveryCondition: Recovery Condition.
get_eval_actions(...): Gets eval actions for TFM trainer.
get_train_actions(...): Gets train actions for TFM trainer.