2021-11-04

GPU 混合精度训练

Pytorch fastai库要实现混合精度计算
只需要在Learner后添加.to_fp16()

learn = Learner(databunch,
                    model,
                    opt_func=optimizer,
                    loss_func=criterion,
                    metrics=[dice, recall_partial, precision_partial]).to_fp16()

亲测有效，时间提升了一半。