huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
910 Commits 74 Branches 16 Tags
a1812f934f4e0830ed3c2f147d13837ccf67f2bd
Commit Graph

2 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
42e1cc8062 Add a batch normalization layer (#508) 
* Add BatchNormalization.

* More batch-norm.

* Add some validation of the inputs.

* More validation.
 2023-08-18 20:05:56 +01:00