Skip to content

Latest commit

 

History

History
22 lines (17 loc) · 2.15 KB

README.md

File metadata and controls

22 lines (17 loc) · 2.15 KB

Mixed Precision Training

Introduction

[OTHERS]

@article{micikevicius2017mixed,
  title={Mixed precision training},
  author={Micikevicius, Paulius and Narang, Sharan and Alben, Jonah and Diamos, Gregory and Elsen, Erich and Garcia, David and Ginsburg, Boris and Houston, Michael and Kuchaiev, Oleksii and Venkatesh, Ganesh and others},
  journal={arXiv preprint arXiv:1710.03740},
  year={2017}
}

Results and Models

Architecture Backbone Style Lr schd Mem (GB) Inf time (fps) box AP mask AP Config Download
Faster R-CNN R-50 pytorch 1x 3.4 28.8 37.5 - config model | log
Mask R-CNN R-50 pytorch 1x 3.6 24.1 38.1 34.7 config model | log
Retinanet R-50 pytorch 1x 2.8 31.6 36.4 config model | log