huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add a convnet training example. (#661)

Browse Source 
* Add a convnet example.

* Dataset fix.

* Randomize batches.
This commit is contained in:
Laurent Mazare
2023-08-29 18:23:01 +01:00
committed by GitHub
parent 71221559d3
commit b31d41e26a
2 changed files with 106 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-datasets/src/vision/mnist.rs
View File

@ -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}")))?;