mmpretrain/resnet50_b32x8_fp16_dynamic_imagenet.py at 4d1fb1a662f8285a7b6850ab64861ff4c8c3887d - mmpretrain - Gitea: Git with a cup of tea

mirrors/mmpretrain

mirror of https://github.com/open-mmlab/mmpretrain.git synced 2025-06-03 14:59:18 +08:00

LXXXXR e76c5a368d

[Feature] Support fp16 training (#178 )

* change mmcls fp16 to mmcv hook

* support fp16

* clean unnessary stuff

2021-03-17 15:53:55 +08:00

5 lines

101 B

Python

Raw Blame History

 _base_ = ['../resnet/resnet50_b32x8_imagenet.py']
 # fp16 settings
 fp16 = dict(loss_scale='dynamic')