diff options
Diffstat (limited to 'candle-examples/examples/mnist-training/main.rs')
| -rw-r--r-- | candle-examples/examples/mnist-training/main.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/mnist-training/main.rs b/candle-examples/examples/mnist-training/main.rs index bcf8677d..2cdad9fa 100644 --- a/candle-examples/examples/mnist-training/main.rs +++ b/candle-examples/examples/mnist-training/main.rs @@ -143,7 +143,7 @@ struct Args { pub fn main() -> anyhow::Result<()> { let args = Args::parse(); // Load the dataset - let m = candle_datasets::vision::mnist::load_dir("data")?; + let m = candle_datasets::vision::mnist::load()?; println!("train-images: {:?}", m.train_images.shape()); println!("train-labels: {:?}", m.train_labels.shape()); println!("test-images: {:?}", m.test_images.shape()); |