mmpretrain/configs/fp16
LXXXXR d3dceddf31
add model zoo for fp16 (#184)
2021-03-25 15:49:26 +08:00
..
README.md add model zoo for fp16 (#184) 2021-03-25 15:49:26 +08:00
resnet50_b32x8_fp16_dynamic_imagenet.py [Feature] Support fp16 training (#178) 2021-03-17 15:53:55 +08:00
resnet50_b32x8_fp16_imagenet.py [Feature] Support fp16 training (#178) 2021-03-17 15:53:55 +08:00

README.md

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

Model Params(M) Flops(G) Mem (GB) Top-1 (%) Top-5 (%) Config Download
ResNet-50 25.56 4.12 1.9 76.32 93.04 config model | log