【PyTorch】解决RuntimeError: adaptive_max_pool2d_backward_cuda …(添加注意力机制)

Error: RuntimeError: adaptive_max_pool2d_backward_cuda does not have a deterministic implementation...

查看log定位error位置:

在scaler.scale(loss).backward()前添加torch.use_deterministic_algorithms(False)

torch.use_deterministic_algorithms(False)  # added
scaler.scale(loss).backward()

本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
THE END
分享
二维码
< <上一篇
下一篇>>