Yolo_Nano icon indicating copy to clipboard operation
Yolo_Nano copied to clipboard

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

Open model-collapse opened this issue 4 years ago • 1 comments

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.

model-collapse avatar May 15 '20 02:05 model-collapse

batch_size is too small

BokyLiu avatar Jul 16 '20 03:07 BokyLiu