hed icon indicating copy to clipboard operation
hed copied to clipboard

issues about the learning rate

Open tengteng95 opened this issue 6 years ago • 2 comments

Thanks for sharing your code and it works well.

I have some doubts about the learning rate. Why learning rate for conv5 is 100x of base learning rate, and classification layer(dsns) is 0.01x of base learning rate? For fune-tuning, learning rate for classification layer(which is trained from scratch) is usually set 10x compared to the backbone network parameters, right?

Besides, if I insert new layers into hed, how should I set the learning rate for them?

tengteng95 avatar Jan 21 '19 07:01 tengteng95

Hi. For the learning rate, I am sorry that I just simply re-use all the settings in the original HED repository and did not tune much about these hyper-parameters. You may change some of them and check effect after 1 or 2 epochs. For the insertion of new layers, you can set 1x at first and check the results. Later modify it to 0.1x / 10x if results are bad.

xwjabc avatar Jan 21 '19 19:01 xwjabc

Many thanks to you! I'd have a try.

tengteng95 avatar Jan 22 '19 02:01 tengteng95