diff options
| author | Taylor Ninesling <taylor.ninesling@gmail.com> | 2024-06-03 23:27:24 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-06-04 06:27:24 +0200 |
| commit | 3f13ad3d79790d02136f4e3ebcd2c8b671615a23 (patch) | |
| tree | 629a24cb4e52e445fffc4e1dd020ade2d224b11b | |
| parent | cd4d941ed10fd334333cf5793e311d2bef88a438 (diff) | |
| download | candle-3f13ad3d79790d02136f4e3ebcd2c8b671615a23.tar.gz candle-3f13ad3d79790d02136f4e3ebcd2c8b671615a23.tar.bz2 candle-3f13ad3d79790d02136f4e3ebcd2c8b671615a23.zip | |
Fix dataset id for MNIST (#2238)
| -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, |