YOLOv8如何关闭AMP混合精度训练?
- 如果你是使用命令行运行的话,只需要在训练参数中添加–unamp即可.
- 如果你是直接代码运行的话,找到这个参数parser.add_argument(‘–unamp’, action=‘store_true’, help=‘Unuse Automatic Mixed Precision (AMP) training’),修改为parser.add_argument(‘–unamp’, action=‘store_true’, default=True, help=‘Unuse Automatic Mixed Precision (AMP) training’)即可.