可以试试将torch.use_deterministic_algorithms(False)添加到train.py里面(反向传播)的这个位置处。我也是参考下面这位博主解决的,如果帮助到大家,并且有时间的话给这位博主点点关注,至于我为什么要发一下呢,就是想着如果多发一遍的话,更多需要这个解决方法的人可能更容易看到。
yolov5改了CBAM注意力机制后报错:runtimeerror:adaptive_avg_pool2d_backward_cuda does not have a deterministic im
最新推荐文章于 2024-03-31 20:57:30 发布