pytorch-YOLOv4 icon indicating copy to clipboard operation
pytorch-YOLOv4 copied to clipboard

Published 20 hours ago • verified publisher icon Tianxiaomo

Training time on coco too long

Open QinghangHong1 opened this issue 3 years ago • 2 comments

Hi, I have tried to train yolov4 on coco2017 dataset with yolov4.cfg using 4 Tesla P100 and I did not change any of the config. It takes me 4.5 hours to train each epoch, which is way too long. What could be wrong in this situation? Thanks a lot!

QinghangHong1 avatar Jun 02 '21 03:06 QinghangHong1

@Tianxiaomo any update on this thread i am also facing the same issues @QinghangHong1 any solutions did u find for this

abhigoku10 avatar Oct 21 '21 05:10 abhigoku10

Hi, You have to use pytorch version 1.4. other newer version will cause a very long training time. I am not sure why we are getting this behavior anyway.

ibrahimsoliman97 avatar Apr 12 '22 04:04 ibrahimsoliman97