How long does it take to train a YoloNano model from zero?

lingtengqiu / Yolo_Nano

Pytorch implementation of yolo_Nano for pedestrian detection

140 stars 15 forks source link

How long does it take to train a YoloNano model from zero? #9

Open model-collapse opened 4 years ago

model-collapse commented 4 years ago

Hi all, I am trying to reproduce the model by the code from this repository. I used a V100 * 1 machine and find the training is very low with a batch_size = 8. Only 1.6 epochs was done after 12 hours. So I would like to know the training time cost of others in order find where the problem is.

BokyLiu commented 4 years ago

batch_size is too small