没了海绵宝宝的派大星 2023-10-24 16:26 采纳率: 0%
浏览 19
已结题

yolov8断点训练后可以更换显卡号吗

yolov8断点训练后,想要继续训练
可以更换显卡号吗,原来用的device=0,1现在想用device=2,3
yolo train resume model=/home/yolov8/runs/detect/train32/weights/last.pt device=2,3 这种方法不好用。

  • 写回答

2条回答 默认 最新

  • 爱晚乏客游 2023-10-24 16:38
    关注

    你要确保你的train32是中的32是最大的数字,否者resume 会自动找到比32更大的数字的文件夹里面去。
    另外就是去train32里面找到配置的yaml文件,默认应该是args.yaml,将里面的device参数手动改为2,3

    img

    然后重新训练看下

    评论

报告相同问题?

问题事件

  • 已结题 (查看结题原因) 11月23日
  • 修改了问题 10月24日
  • 创建了问题 10月24日