Mixed Precision Training
Introduction
@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 |
