yolo系列多卡训练命令

yolo系列多卡训练

多卡训练命令示例

python train.py  --batch 64 --data coco.yaml --weights yolov5s.pt --device 0,1

参数

  • –batch:指定GPU一次处理的样本数量,即批量
  • –data:指定描述训练和验证数据集位置,以及标签类别的配置文件
  • –weights:指定预训练权重,如果指定了预训练权重那么就不用从0开始训练,这样有助于迁移学习
  • –device:指定GPU索引号,有几块卡,就可以指定几个数字,比如如果4张卡同时训练,那就是0,1,2,3这样写

你可能感兴趣的:(深度学习,YOLO,深度学习,机器学习)