Skip to main content

ml_training_basic_usage/
basic_usage.rs

1use use_ml_training::{MlBatchSize, MlLearningRate, MlOptimizerKind, MlTrainingRunId};
2
3fn main() -> Result<(), use_ml_training::MlTrainingError> {
4    let run_id = MlTrainingRunId::new("run-001")?;
5    let batch_size = MlBatchSize::new(32)?;
6    let learning_rate = MlLearningRate::new(0.001)?;
7    let optimizer: MlOptimizerKind = "adamw".parse()?;
8
9    assert_eq!(run_id.as_str(), "run-001");
10    assert_eq!(batch_size.get(), 32);
11    assert_eq!(learning_rate.value(), 0.001);
12    assert_eq!(optimizer, MlOptimizerKind::AdamW);
13    Ok(())
14}