search

yunxiaoshi / Neural-IMage-Assessment

A PyTorch Implementation of Neural IMage Assessment
Other
508 stars 92 forks source link

Can you share the settings of Learning Rate? #2

Closed ken-ouyang closed 5 years ago

ken-ouyang commented 6 years ago

Hi,

I have tried the default learning rate settings and it does not converge. Can you share the training settings?

Many thanks.

yunxiaoshi commented 6 years ago

Hi @ken-ouyang

I used an lr of 0.001 for the conv base and 0.01 for the dense block, for which I got a 0.22 EMD loss. But I suspect they are not optimal and can be further improved.