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