Module: tfm.core.actions

Provides TFM orbit actions and associated helper functions/classes.

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.

Functions

get_eval_actions(...): Gets eval actions for TFM trainer.

get_train_actions(...): Gets train actions for TFM trainer.