load_checkpoint

numrs::ops::model

Function load_checkpoint 

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

Load training checkpoint

§Arguments

  • path - Path to load checkpoint from

§Returns

Tuple of (model, training_state)