0
点赞
收藏
分享

微信扫一扫

pytorch中的dataset和datalaoder

ixiaoyang8 2022-01-24 阅读 72

1. 使用Dataset数据集

import torchvision
使用dataset数据集需要设置的参数
使用方法:

# 1. 如何使用dataset标准数据集
train_set = torchvision.datasets.CIFAR10(root="./dataset2", train=True, transform=dataset_transform, download=True)
test_set = torchvision.datasets.CIFAR10(root="./dataset2", train=False, transform=dataset_transform, download=True)
# 获取img和target
img, target = test_set[i]

DataLoader

from torch.utils.data import DataLoader
常用参数:
dataset
batch_size:每次取数据集的大小
shuffle:每次循环取是否打乱。True为打乱(常用),False为不打乱。
num_workers
drop_last:剩余的数据集是否保留。True为保留,False为不保留

test_loader = DataLoader(dataset=test_data, batch_size=64, shuffle=True, num_workers=0, drop_last=False)
举报

相关推荐

0 条评论