huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 04:10:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
1d1d6d4fe6dbeee03e3a7ca40b03a555ce145ff6
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.