summaryrefslogtreecommitdiff
path: root/candle-nn/examples/basic_optimizer.rs
Commit message (Collapse)AuthorAgeFilesLines
* Add the optimizer trait. (#702)Laurent Mazare2023-09-011-1/+1
|
* Add a yolo-v3 example. (#528)Laurent Mazare2023-08-201-0/+6
| | | | | | | | | | | | | | | * Add a couple functions required for yolo. * Add the yolo-v3 example. * Add minimum and maximum. * Use the newly introduced maximum. * Cuda support for min/max + add some testing. * Allow for more tests to work with accelerate. * Fix a typo.
* Add a simple Module trait and implement it for the various nn layers (#500)Laurent Mazare2023-08-181-1/+1
| | | | | | | * Start adding the module trait. * Use the module trait. * Implement module for qmatmul.
* Add a basic optimizer example. (#454)Laurent Mazare2023-08-151-0/+33