huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-22 12:28:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
915 Commits 74 Branches 16 Tags
4300864ce928cc901818771e5e1bb9202b96ede5
Commit Graph

2 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
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