EfficientNet-PyTorch icon indicating copy to clipboard operation
EfficientNet-PyTorch copied to clipboard

Published 20 hours ago • verified publisher icon lukemelas

the optimizer of training net

Open shenwenxiao opened this issue 4 years ago • 0 comments

In examples/imagenet/main.py, the optimizer of training is set as SGD. But in the paper, the optimizer of training is set as RMSProp. I want to train EfficientNet but do not know how to select optimizer. Could you explain it for me?

shenwenxiao avatar Jan 19 '21 14:01 shenwenxiao