Open gedance opened 3 years ago
I will answer my personal thoughts for similar reasons. My guess might be that it is a model that is too light for the problem to be solved. (Actually, I can't find any other classes other than faces because of a similar problem) If you are using a pretrained model, why not try sgd, not Adam?
Hi @gedance, Can you help me train the model on a custom dataset (data preparation from YOLO format to train format)? I encountered several different errors and could not resolve them.
Hello,
I've been facing overfitting issues when I train with my own datasets. I'm doing some researches for detecting vehicle plates, and there are over 60K images of vehicles, no problem with annotations. There's only one vehicle in each image, rarely two.
My issues are
overfitting happens as learning rate decrease as attached
RFB setting --milestones 150, 220 --batch-size 24 --input size 320
running cuda
lr_rate: 0.01 decrease at epoch 150 to 0.001, then 0.0001 at epoch 220
I've tried adding dropouts, using Adam optimizer, adding augmentations, but didn't help. does anyone know what else I can try? Early stopping is the only answer in my case?
I am a new to ML, your experienced advice would be very appreciated.
Thanks!