diff options
Diffstat (limited to 'candle-datasets/src/vision/mnist.rs')
| -rw-r--r-- | candle-datasets/src/vision/mnist.rs | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/candle-datasets/src/vision/mnist.rs b/candle-datasets/src/vision/mnist.rs index 1085edd6..30b0d01f 100644 --- a/candle-datasets/src/vision/mnist.rs +++ b/candle-datasets/src/vision/mnist.rs @@ -101,10 +101,10 @@ pub fn load() -> Result<crate::vision::Dataset> { ); let repo = api.repo(repo); let test_parquet_filename = repo - .get("mnist/mnist-test.parquet") + .get("mnist/test/0000.parquet") .map_err(|e| Error::Msg(format!("Api error: {e}")))?; let train_parquet_filename = repo - .get("mnist/mnist-train.parquet") + .get("mnist/train/0000.parquet") .map_err(|e| Error::Msg(format!("Api error: {e}")))?; let test_parquet = SerializedFileReader::new(std::fs::File::open(test_parquet_filename)?) .map_err(|e| Error::Msg(format!("Parquet error: {e}")))?; |