Skip to main content

load_training_checkpoint

yscv_model

Function load_training_checkpoint 

Source 
pub fn load_training_checkpoint(
    path: &Path,
) -> Result<(HashMap<String, Tensor>, HashMap<String, Tensor>), ModelError>
Expand description

Load a full training checkpoint, splitting model weights from optimizer state.

Returns (model_weights, optimizer_state).