training time is too long

GOATmessi8 / ASFF

yolov3 with mobilenet v2 and ASFF

GNU General Public License v3.0

1.05k stars 216 forks source link

training time is too long #54

Open amsword opened 4 years ago

amsword commented 4 years ago

By default, 300 epochs are used for the training. On a machine with 4 P100, it needs about 21 days. Is it normal? How is the training time with V100 as described in Readme?

GOATmessi8 commented 4 years ago

@amsword About 6 days on 4 V100, with 64 batchsize. I have not tried to optimize many training parameters such as epochs and lr, and I think 200 epochs are enough to get a good result.