torchvision/models/detection/roi_heads.py · 14af9de63dac730c3240a83fa963d4e91bc562a4 · OpenDAS / vision

Replace L1 loss with beta smooth L1 loss to achieve better performance (#2113) · 8df3f29a

Potter Hsu authored Apr 25, 2020

* For detection model, replace L1 loss with beta smooth L1 loss to achieve better performance.

* Add type annotations for torchscript

* Resolve E226

8df3f29a

roi_heads.py 32.8 KB

Replace roi_heads.py