summaryrefslogtreecommitdiff
path: root/candle-nn/src/optim.rs
Commit message (Expand)AuthorAgeFilesLines
* Expose AdamW parameters (#1449)Dave Lage2023-12-161-0/+8
* Only optimize float tensors. (#1069)Laurent Mazare2023-10-101-0/+5
* Add the optimizer trait. (#702)Laurent Mazare2023-09-011-46/+60
* Set the learning rate. (#687)Laurent Mazare2023-08-311-0/+8
* Add the AdamW optimizer. (#307)Laurent Mazare2023-08-021-2/+114
* Llama more training (#297)Laurent Mazare2023-08-011-1/+8
* Add the gradient for reduce-sum. (#162)Laurent Mazare2023-07-131-1/+1
* Add the SGD optimizer (#160)Laurent Mazare2023-07-131-0/+47