MNIST

dset.MNIST(root, train=True, transform=None, target_transform=None, download=False)

参数说明:

  • root : processed/training.pt 和 processed/test.pt 的主目录
  • train : True = 训练集, False = 测试集
  • download : True = 从互联网上下载数据集,并把数据集放在root目录下. 如果数据集之前下载过,将处理过的数据(minist.py中有相关函数)放在processed文件夹下。

results matching ""

    No results matching ""