from apex import amp model, optimizer = amp.initialize(model, optimizer, opt_level="O1") # 这里是“欧一”,不是“零一” with amp.scale_loss(loss, optimizer) as scaled_loss: scaled_loss.backward()
命令行
python tools/train.py -f exps/example/yolox_voc/yolox_voc_s.py -d 8 -b 64 --fp16 -o -c /path/to/yolox_s.pth
Windows——PyTorch上实现混合精度训练加速Apex
于 2023-02-28 20:44:37 首次发布