nfflr.train.trainer.setup_checkpointing

nfflr.train.trainer.setup_checkpointing#

nfflr.train.trainer.setup_checkpointing(state: dict, config: TrainingConfig)[source]#

Configure model and trainer checkpointing.

state should contain at least model, optimizer, and trainer.