search

AILab-CVC / YOLO-World

[CVPR 2024] Real-Time Open-Vocabulary Object Detection
https://www.yoloworld.cc
GNU General Public License v3.0
4.64k stars 450 forks source link

关于训练 #116

Closed dikapiliao1 closed 8 months ago

dikapiliao1 commented 8 months ago

请问4块GPU能训练吗

wondervictor commented 8 months ago

可以训，不过我建议finetune，pre-train的话开销会很大

dikapiliao1 commented 8 months ago

估计要多久呢~

wondervictor commented 8 months ago

4卡如果是fine-tune COCO的话，80个epoch 应该在1天内，但是如果是pre-train模型的话，应该比较长（目前32卡需要2天时间）。如果你实在需要修改pre-trained模型，可以下载我们提供的weights，然后在pre-train data上继续微调模型，这个比较有效，我们YOLO-World-v2-L-1280大尺寸输入也是采用继续微调的方式去实现的，性能会更好。

dikapiliao1 commented 8 months ago

好的 谢谢~

wondervictor commented 8 months ago

Thanks for your interest. If you have any questions about YOLO-World in the future, you're welcome to open a new issue.