这是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