Windows——PyTorch上实现混合精度训练加速Apex

失梦丧犬

已于 2023-02-28 21:01:17 修改

阅读量233

点赞数

文章标签： pytorch 深度学习机器学习

于 2023-02-28 20:44:37 首次发布

本文链接：https://blog.youkuaiyun.com/qq_51623862/article/details/129269567

版权

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