diff options
| -rw-r--r-- | candle-datasets/src/vision/mnist.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-datasets/src/vision/mnist.rs b/candle-datasets/src/vision/mnist.rs index 2dac883c..eb79e17e 100644 --- a/candle-datasets/src/vision/mnist.rs +++ b/candle-datasets/src/vision/mnist.rs @@ -89,7 +89,7 @@ fn load_parquet(parquet: SerializedFileReader<std::fs::File>) -> Result<(Tensor, pub fn load() -> Result<crate::vision::Dataset> { let api = Api::new().map_err(|e| Error::Msg(format!("Api error: {e}")))?; - let dataset_id = "mnist".to_string(); + let dataset_id = "ylecun/mnist".to_string(); let repo = Repo::with_revision( dataset_id, RepoType::Dataset, |