huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 19:18:50 +00:00
Code Issues Packages Projects Releases Wiki Activity
601 Commits 74 Branches 16 Tags
43c72232927ca80c850a73ce977c2063d5a2dcf5
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
57be3638d8 Add the pytorch version of the linear regression as a comment. (#163) 
* Add the pytorch version of the linear regression.

* Typo.
 2023-07-13 21:05:57 +01:00
Laurent Mazare
23e105cd94 Add the gradient for reduce-sum. (#162) 
* Add the gradient for reduce-sum.

* And add the gradient for the broadcast ops.

* Add some backprop tests.

* Add some linear regression example.
 2023-07-13 20:14:10 +01:00
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