yolov8添加注意力机制报错
UserWarning: adaptive_avg_pool2d_backward_cuda does not have a deterministic implementation, but you set ‘torch.use_deterministic_algorithms(True, warn_only=True)’. 出现这个问题的根本原因是PyTorch中的某些操作不支持确定性行为,而你已经设置了使用确定性算法。
解决方法:找到ultralytics/engine/trainer.py文件,修改backward 390行左右:
torch.use_deterministic_algorithms(False) # 禁用确定性算法模式,添加这个
self.scaler.scale(self.loss).backward()