diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-07-13 19:05:44 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-07-13 19:05:44 +0100 |
| commit | ded93a116983da7c84ea224a6191bcbc3a7fdef1 (patch) | |
| tree | d0fcada3d4a2bdc7fab269b5758d0723f3d077c0 /candle-nn/src/embedding.rs | |
| parent | 5ee3c95582b6d199cbad42808ba56beab234f271 (diff) | |
| download | candle-ded93a116983da7c84ea224a6191bcbc3a7fdef1.tar.gz candle-ded93a116983da7c84ea224a6191bcbc3a7fdef1.tar.bz2 candle-ded93a116983da7c84ea224a6191bcbc3a7fdef1.zip | |
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.
Diffstat (limited to 'candle-nn/src/embedding.rs')
0 files changed, 0 insertions, 0 deletions