【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
二维码