May I know your hyper-parameters of "ssd_lite_mobilenetv1"

ShuangXieIrene / ssds.pytorch

Repository for Single Shot MultiBox Detector and its variants, implemented with pytorch, python3.

MIT License

568 stars 165 forks source link

May I know your hyper-parameters of "ssd_lite_mobilenetv1" #58

Closed haoyango closed 5 years ago

haoyango commented 5 years ago

First,thanks for your great work! Now,I have trained "ssd_lite_mobilenetv1" on voc data for several times,but the results are not so good,I found the loss curve fluctuate a little violently,so May I know your learning-rate ,weight_decay and warm_up_epochs?

foreverYoungGitHub commented 5 years ago

Hi, the detial of the hyperparameter is in this file: https://github.com/ShuangXieIrene/ssds.pytorch/blob/master/experiments/cfgs/ssd_lite_mobilenetv1_train_voc.yml

Let me know if you have any other questions.