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
932 Commits 74 Branches 16 Tags
d4968295a0b75f1adafcc19a7cfe15e8acf3d58d
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
a1812f934f Add a yolo-v3 example. (#528) 
* 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.
 2023-08-20 18:19:37 +01:00
Laurent Mazare
0902846f25 Add the AdamW optimizer. (#307) 
* Add the AdamW optimizer.

* Add some AdamW test validated against PyTorch.
 2023-08-02 14:03:49 +01:00
Laurent Mazare
3eb2bc6d07 Softmax numerical stability. (#267) 
* Softmax numerical stability.

* Fix the flash-attn test.
 2023-07-28 13:13:01 +01:00