YOLOv8如何关闭AMP混合精度训练?

  1. 如果你是使用命令行运行的话,只需要在训练参数中添加–unamp即可.
  2. 如果你是直接代码运行的话,找到这个参数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’)即可.

你可能感兴趣的:(yolov8改进,YOLO,前端,linux)