search

GuoQuanhao / yolov5-Paddle

YOLOv5🚀 reproduction by Guo Quanhao using PaddlePaddle
41 stars 16 forks source link

多卡训练 #3

Closed thunder95 closed 1 year ago

thunder95 commented 2 years ago

浩佬有多卡训练的脚本么， 虽然python -m paddle.distributed.launch train.py也能跑，但是打印日志这些有问题

GuoQuanhao commented 2 years ago

多卡还没写哦，最近忙着找实习，估计代码更新得延期一会儿

GuoQuanhao commented 1 year ago

本次更新实现了yolov5的全新迁移，能够实现分类、分割、检测和各种格式模型的导出。现在已支持多卡了，以及任意指定单卡