Created by: littletomatodonkey

You can add the following config to spport the mixed precision training.

use_fp16: True
amp_scale_loss: 128.0
use_dynamic_loss_scaling: True

More detailed configs is avaiable in configs/ResNet/ResNet50_fp16.yml