YOLOv10 训练时loss是nan,训练时显示页面太小
·
找到下载的ultralytics包中的trainer,打开
找到 # Check AMP下的代码
将self.amp = bool(self.amp) # as boolean
注释掉
在下一行增加self.amp = False即可。
当显示页面太小时
将workers设置为0即可
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐


所有评论(0)