Home

Awesome

Pytorch Squeeznet

Pytorch implementation of Squeezenet model as described in https://arxiv.org/abs/1602.07360 on cifar-10 Data.

The definition of Squeezenet model is present model.py. The training procedure resides in the file main.py

Command to train the Squeezenet model on CIFAR 10 data is:

python main.py --batch-size 32 --epoch 10

Other options which can be used are specified in main.py Eg: if you want to use a pretrained_model

python main.py --batch-size 32 --epoch 10 --model_name "pretrained model"

I am currently using SGD for training : learning rate and weight decay are currently updated using a 55 epoch learning rule, this usually gives good performance, but if you want to use something of your own, you can specify it by passing learning_rate and weight_decay parameter like so

python main.py --batch-size 32 --epoch 10 --learning_rate 1e-3 --epoch_55