1.yolov5 7.0 训练时报错
RuntimeError: adaptive_max_pool2d_backward_cuda does not have a deterministic implementation, but you set ‘torch.use_deterministic_algorithms(True)’
2.解决方法
train.py 文件中找到如下代码
scaler.scale(loss).backward()
如图在此代码前添加如下代码:
torch.use_deterministic_algorithms(False)
即可解决!!!!