save_checkpoint

numrs::ops::model

Function save_checkpoint 

Source 
pub fn save_checkpoint(
    model: &OnnxModel,
    training_state: &TrainingState,
    path: &str,
) -> Result<()>
Expand description

Save training checkpoint

ยงArguments

  • model - The model to save
  • training_state - Current training state (epoch, loss, optimizer state)
  • path - Path to save checkpoint