Back to Candle

candle-mnist-training

candle-examples/examples/mnist-training/README.md

0.10.1362 B
Original Source

candle-mnist-training

Training a 2 layer MLP on mnist in Candle.

Running an example

bash
$ cargo run --example mnist-training --features candle-datasets

> train-images: [60000, 784]
> train-labels: [60000]
> test-images: [10000, 784]
> test-labels: [10000]
>    1 train loss:  2.30265 test acc: 68.08%
>    2 train loss:  1.50815 test acc: 60.77%