from apex import amp
model, optimizer = amp.initialize(model, optimizer, opt_level='O1') # not 01
with amp.scale_loss(loss, optimizer) as scaled_loss:
scaled_loss.backward
工作笔记|基于Apex的混合精度加速
最新推荐文章于 2022-04-17 17:06:20 发布
from apex import amp
model, optimizer = amp.initialize(model, optimizer, opt_level='O1') # not 01
with amp.scale_loss(loss, optimizer) as scaled_loss:
scaled_loss.backward