summaryrefslogtreecommitdiff
path: root/candle-examples/examples/mnist-training/main.rs
Commit message (Expand)AuthorAgeFilesLines
* Allow for different behavior between training and eval (#1213)Laurent Mazare2023-10-291-2/+2
* Add the optimizer trait. (#702)Laurent Mazare2023-09-011-2/+2
* Mnist training dropout (#677)Laurent Mazare2023-08-301-7/+11
* Add some documentation. (#673)Laurent Mazare2023-08-301-1/+1
* Simplify usage of the pool functions. (#662)Laurent Mazare2023-08-291-7/+9
* Add a convnet training example. (#661)Laurent Mazare2023-08-291-1/+104
* Re-enable local dir for mnist.Nicolas Patry2023-08-281-1/+9
* Training:Nicolas Patry2023-08-281-1/+1
* VarBuilder cleanup (#627)Laurent Mazare2023-08-271-2/+2
* Add a yolo-v3 example. (#528)Laurent Mazare2023-08-201-0/+3
* Add a simple Module trait and implement it for the various nn layers (#500)Laurent Mazare2023-08-181-1/+1
* Add the candle-datasets crate (#322)Laurent Mazare2023-08-051-2/+2
* Llama more training (#297)Laurent Mazare2023-08-011-124/+15
* Make the nll op closer to the pytorch version + add a test. (#286)Laurent Mazare2023-07-311-4/+1
* Add a flag to set the number of epochs in the mnist training (#283)Laurent Mazare2023-07-311-14/+28
* Load a trained checkpoint in the mnist example. (#280)Laurent Mazare2023-07-301-3/+36
* Add a flag to save the trained weights. (#279)Laurent Mazare2023-07-301-0/+228