huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
7ec4f64d38126f9d0c43e3dd10fe04bc52c8b6ac
candle/candle-book/src/training/mnist.md
Nicolas Patry dd06d93d0b Cleanup: 
- Moved around book from `examples` to `candle-book` proper (overlapping
  the book and the lib structures)
2023-08-28 15:15:26 +02:00

336 B
Raw Blame History

MNIST

So we now have downloaded the MNIST parquet files, let's put them in a simple struct.

{{#include ../lib.rs:book_training_3}}

The parsing of the file and putting it into single tensors requires the dataset to fit the entire memory. It is quite rudimentary, but simple enough for a small dataset like MNIST.