Skip to content

chenhang98/mobileNet-v2_cifar10

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

mobileNet-v2_cifar10

a pytorch implement of mobileNet v2 on cifar10

architecture

  • The origin mobileNet architecture is designed specifically for ImageNet where images' size is 224x224x3. To make it fit cifar10's size (32x32x3), I have disabled some downsample layer, i.e. replace the first few layers which have stride 2 with stride 1, as highlighted below.

    table

training

  • Run python3 train.py to start training

  • Run python3 plot.py to show training curve

  • Or you can run python3 train_ddp.py using ddp and cosine scheduler(final validacc = 94.71%)

  • I have trained this model (width multiplier = 1, more setups can be seen in train.py) on two titan x, which takes about 6 hours, the weights and logs are available in folder bak

result

  • The model can achieve max / mean accuracy 94.69% / 94.52% on validation set. Here the "mean accuracy" refers to mean of last 10 accuracy.

  • accuracy & loss - iterations curves are shown below:

    curve

dependency

  • python 3+
  • pytorch 0.4.0

About

a pytorch implement of mobileNet v2 on cifar10

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages