summaryrefslogtreecommitdiff
path: root/candle-nn/src/batch_norm.rs
Commit message (Expand)AuthorAgeFilesLines
* Detach the tensors on batch-norm eval. (#1702)Laurent Mazare2024-02-131-2/+12
* Do not implement Module for BatchNorm. (#1513)Laurent Mazare2024-01-011-13/+13
* Small tweaks to batch-norm. (#1505)Laurent Mazare2023-12-301-19/+16
* [Breaking] Add training to batchnorm with exponential moving average (#1504)nkoppel2023-12-301-50/+158
* Add fuse-conv-bn method for Conv2d (#1196)jamjamjon2023-10-271-0/+4
* Expose the fields from batch-norm. (#1176)Laurent Mazare2023-10-251-2/+12
* Add clone to various nn layers. (#910)Laurent Mazare2023-09-201-1/+1
* VarBuilder cleanup (#627)Laurent Mazare2023-08-271-4/+4
* Some fixes for yolo-v3. (#529)Laurent Mazare2023-08-201-4/+49
* Add a batch normalization layer (#508)Laurent Mazare2023-08-181-0/+154