鹿酸梅 发表于 2022-5-24 16:19:55

请大神帮忙解决下这两段代码在划分数据集时候的使用区别

前提是我已经下载好了CIFAR10数据集。
第一种方法:
train_set = torchvision.datasets.CIFAR10(root='./data', train=True,
                                             download=False, transform=transform)

    train_loader = torch.utils.data.DataLoader(train_set, batch_size=36,
                                             shuffle=True, num_workers=0)

    val_set = torchvision.datasets.CIFAR10(root='./data', train=False,
                                           download=True, transform=transform)
    val_loader = torch.utils.data.DataLoader(val_set, batch_size=5000,
                                             shuffle=False, num_workers=0)

第二种方法:
cifar10_dir = 'F:\Jupyter\CIFAR10'
X_train, y_train, X_test, y_test = load_CIFAR10(cifar10_dir)


我看大多数帖子都是用的第一种方法,但是搞不清这两种方法的区别,是不是因为第一种更好呢?球球大神帮忙解答一下
页: [1]
查看完整版本: 请大神帮忙解决下这两段代码在划分数据集时候的使用区别