mmpretrain/configs/resnet/resnet50_8xb32-fp16_in1k.py

5 lines
84 B
Python

_base_ = ['./resnet50_8xb32_in1k.py']
# fp16 settings
fp16 = dict(loss_scale=512.)