这是pytorch初学者的游乐场,其中包含流行数据集上的预定义模型。 目前我们支持
mnist,svhn
cifar10,cifar100
stl10
亚历克斯网
vgg16,vgg16_bn,vgg19,vgg19_bn
resnet18,resnet34,resnet50,resnet101,resnet152
squeezenet_v0,squeezenet_v1
inception_v3
这是MNIST数据集的示例。 这将自动下载数据集和预先训练的模型。
import torch
from torch.autograd import Variable
from utee import selector
model_raw, ds_fetcher, is_imagenet = selector.select('mnist')
ds_val = ds_fetcher(b
1