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

how to train? #14

Closed isalirezag closed 5 years ago

isalirezag commented 6 years ago

Can you please tell us how can we train the model from scratch? and also how can we test it?

AIROBOTAI commented 6 years ago

Same question. Unless we know exactly the hyperparameters used by the author, we cannot use the code at all.

basaltzhang commented 5 years ago

I move the MobilenetV2.py into this project. https://github.com/miraclewkf/MobileNetV2-PyTorch lr = 0.045, gamma=0.92, epoch = 99. batchsize = 320 After training one epoch the validation accuracy is 0.7207