TORCH.UTILS.DATA

DataLoader : dataset, sampler, dataset 에서 iterable 제공

CLASS torch.utils.data.DataLoader(dataset, batch_size=1, shuffle=None)

loader = DataLoader(dataset, batch_size=2)

for batch_idx, sample in enumerate(loader):
	print(sample)

 

Dataset 종류

• Map-sytle datasets : __getitem__(), __len__()
• Iterable-style datasets : __iter__()