Fix dataset id for MNIST (#2238)

huggingface · Jun 4, 2024 · 3f13ad3 · 3f13ad3
1 parent cd4d941
commit 3f13ad3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,