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
965 Commits 74 Branches 16 Tags
c265ac50fae267464b5eca402ffbe8ccdf75d68d
Commit Graph

3 Commits

Author SHA1 Message Date
Laurent Mazare
11c7e7bd67 Some fixes for yolo-v3. (#529) 
* Some fixes for yolo-v3.

* Use the running stats for inference in the batch-norm layer.

* Get some proper predictions for yolo.

* Avoid the quadratic insertion.
 2023-08-20 23:19:15 +01:00
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