Skip to main content

train

Function train 

Source
pub fn train(
    spec: &VocabularySpace,
    examples: &[TaggerExample],
    cfg: &TrainConfig,
) -> Result<(VarMap, TrainReport, Vec<String>), String>
Expand description

Finetune Head A + Head B. Returns the report; the trained weights are left in varmap for saving.