1.在ultralytics-main\ultralytics\yolo\cfg\ default.yaml文件中,找到amp参数,将其设置为False
2.ultralytics/engine/validator.py的115行,注释掉
#self.args.half = self.device.type != “cpu” # force FP16 val during training
3.ultralytics/engine/trainer.py中,搜索self.amp:
self.amp = torch.tensor(self.args.amp).to(self.device) # True or False
改为 self.amp=false
4.加大数据集、训练轮数epoch
(我忘记之前是怎么样的,现在使用我的笔记本跑的话,coco128的都是0。但是如果换成大的数据集–试过2000多张的,就恢复正常了)
阿克苏地区网站建设_网站建设公司_跨域_seo优化