yjh0410 / RT-ODLab

YOLO Tutorial
MIT License
156 stars 32 forks source link

请问train_ddp.sh文件在哪,想要尝试一下多卡训练 #2

Open lisenjie757 opened 1 year ago

lisenjie757 commented 1 year ago

以及这是博主最新的yolo仓库吗,这个仓库是不是要比博主的PyTorch_YOLO-Family仓库要好

yjh0410 commented 1 year ago

@lisenjie757 已上传train_ddp.sh文件,不过受限于我贫瘠的计算资源,我没尝试过多卡训练,所以不知道会不会有bug。这个是我目前最新的YOLO仓库,是为了配合后续要出版的图书而重新写的,要比以往我的YOLO仓库要好很多,不论是代码规范、还是技术细节以及型嫩,都要好很多。