运行yolov5融合注意力机制的代码时,出现错误报错代码如下
请博主们答疑解惑
收起
在train.py文件中的scaler.scale(loss).backward()前添加torch.use_deterministic_algorithms(False)这样就改好啦
报告相同问题?