梨涡的漫漫求学路 2023-04-04 12:51 采纳率: 50%
浏览 14
已结题

运行yolov5代码时出现报错的问题

运行yolov5融合注意力机制的代码时,出现错误
报错代码如下

img

请博主们答疑解惑

  • 写回答

3条回答 默认 最新

  • 梨涡的漫漫求学路 2023-04-10 13:30
    关注

    在train.py文件中的scaler.scale(loss).backward()前添加torch.use_deterministic_algorithms(False)
    这样就改好啦

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(2条)

报告相同问题?

问题事件

  • 系统已结题 4月18日
  • 已采纳回答 4月10日
  • 创建了问题 4月4日