search

tonylins / pytorch-mobilenet-v2

A PyTorch implementation of MobileNet V2 architecture and pretrained model.
Apache License 2.0
1.37k stars 328 forks source link

the problem with RMSprop #4

Open FatherOfHam opened 6 years ago

FatherOfHam commented 6 years ago

something seems wrong with the loss of RMSprop. Any advice?

huangbiubiu commented 4 years ago

I believe there are something different in tf RMSpop and PyTorch RMSProp. https://github.com/marvis/pytorch-mobilenet reports bad RMSprop result on MobileNet v1.