Training:

- Removed a lot of surface (SerializedFileReader ownership is really
  painful).
- Moved example + vision to hf.co version.
- Removed feature gate.
This commit is contained in:
Nicolas Patry
2023-08-14 17:23:08 +02:00
parent dd02f589c0
commit d7a273be51
7 changed files with 148 additions and 16 deletions

View File

@ -143,7 +143,7 @@ struct Args {
pub fn main() -> anyhow::Result<()> {
let args = Args::parse();
// Load the dataset
let m = candle_datasets::vision::mnist::load_dir("data")?;
let m = candle_datasets::vision::mnist::load()?;
println!("train-images: {:?}", m.train_images.shape());
println!("train-labels: {:?}", m.train_labels.shape());
println!("test-images: {:?}", m.test_images.shape());