如果使用更大的数据集？例如CIFAR100？

[longmalongma]

感谢你很棒的工作，如果使用更大的数据集？例如CIFAR100？能提供可以跑多个不同数据集的训练脚本吗？谢谢

[AlexHex7]

@longmalongma Hi,

torchvision.datasets中提供了常见的数据集(CIFAR, VOC, Cityscapes等)。

可修改代码中的data部分:

Non-local_pytorch/demo_MNIST_train.py

Lines 9 to 14 in 57f5230

	train_data = torchvision.datasets.MNIST(root='./mnist', train=True,
	transform=torchvision.transforms.ToTensor(),
	download=True)
	test_data = torchvision.datasets.MNIST(root='./mnist/',
	transform=torchvision.transforms.ToTensor(),
	train=False)

例如，修改为CIFAR100, 并设计合适输入尺寸的Network.
train_data = torchvision.datasets.CIFAR100(...)