mmsegmentation/configs/fp16/pspnet_r101-d8_512x1024_80k_fp16_cityscapes.py
Jerry Jiarui XU e1f5b9231e Support FP16 (#21)
* Support FP16

* add miss folder

* add tests

* remove useless config

* update memory

* reduce config

* migrate fp16 to mmcv

* add model link
2020-07-20 15:17:18 +08:00

4 lines
146 B
Python

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