Updating example mnist

delta-rs · Nov 26, 2024 · d72c3b2 · d72c3b2
1 parent 9c847d1
commit d72c3b2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/examples/mnist/src/main.rs b/examples/mnist/src/main.rs
@@ -22,6 +22,9 @@ async fn main() {
     let train_data = MnistDataset::load_train().await;
     let test_data = MnistDataset::load_test().await;
 
+    println!("Training the model...");
+    println!("Train data size: {}", train_data.len());
+
     model.fit(&train_data, 10, 32);
 
     // Evaluate the model