mmsegmentation/configs/fp16/deeplabv3_r101-d8_512x1024_80k_fp16_cityscapes.py
Jerry Jiarui XU 89f8a6a410
[Bug fix] fixed fp16 inference (#497)
* fixed fp16

* update fps
2021-04-24 09:57:33 -07:00

6 lines
185 B
Python

_base_ = '../deeplabv3/deeplabv3_r101-d8_512x1024_80k_cityscapes.py'
# fp16 settings
optimizer_config = dict(type='Fp16OptimizerHook', loss_scale=512.)
# fp16 placeholder
fp16 = dict()