马上注册,结交更多好友,享用更多功能^_^
您需要 登录 才可以下载或查看,没有账号?立即注册
x
前提是我已经下载好了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)
我看大多数帖子都是用的第一种方法,但是搞不清这两种方法的区别,是不是因为第一种更好呢?球球大神帮忙解答一下 |