Training time on coco too long

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4

Apache License 2.0

4.47k stars 1.49k forks source link

Training time on coco too long #436

Open QinghangHong1 opened 3 years ago

QinghangHong1 commented 3 years ago

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!

abhigoku10 commented 2 years ago

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

ibrahimsoliman97 commented 2 years ago

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.