huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
536 Commits 74 Branches 16 Tags
3c02ea56b0dcc39c30ad0d41d942384cc28f65c2
Commit Graph

2 Commits

Author SHA1 Message Date
Laurent Mazare
ded93a1169 Add the SGD optimizer (#160) 
* Add the nn::optim and some conversion traits.

* Add the backward_step function for SGD.

* Get the SGD optimizer to work and add a test.

* Make the test slighly simpler.
 2023-07-13 19:05:44 +01:00
Laurent Mazare
71cd3745a9 Add some layer-norm tests. (#121) 2023-07-10 14:43:04 +01:00