search

aaron-xichen / pytorch-playground

Base pretrained models and datasets in pytorch (MNIST, SVHN, CIFAR10, CIFAR100, STL10, AlexNet, VGG16, VGG19, ResNet, Inception, SqueezeNet)
MIT License
2.62k stars 612 forks source link

Shouldn't it be m = vgg.vgg16_bn(model_root) ?? #44

Closed AndyMoriaty closed 4 years ago

AndyMoriaty commented 4 years ago

https://github.com/aaron-xichen/pytorch-playground/blob/713e29b2c4324e95685fe3641036fe32261524f9/utee/selector.py#L77

aaron-xichen commented 4 years ago

fixed, thanks so much~