pytorch_models

file

model.py contain AlexNet, VGGNet16, Resnet18, Resnet34, Resnet50

train_model.py is use to train model

RNN.py is RNN train and test demo

params

in_channels=3 class_num=1000

simplenet ('Total number of params', 928070) ('Total layers', 4)

alexnet ('Total number of params', 8159464) ('Total layers', 7)

vggnet16 ('Total number of params', 6255496) ('Total layers', 15)

resnet18 ('Total number of params', 13070568) ('Total layers', 21)

resnet34 ('Total number of params', 23182440) ('Total layers', 37)

resnet50 ('Total number of params', 25583592) ('Total layers', 54)

Train

simplenet epoch=20 test_correct=0.8555

RNN

RNN.py