Skip to main content

train_sgd

Function train_sgd 

Source
pub fn train_sgd(
    dataset: &SyntheticDataset,
    model_cfg: KstemitConfig,
    train_cfg: TrainConfig,
) -> TrainResult
Expand description

Train a new model on a synthetic dataset.