Commit aa9fa44a authored by Alykhan Tejani's avatar Alykhan Tejani Committed by Francisco Massa
Browse files

remove test/cifar.py (#297)

parent 901c1ad2
import torch
import torchvision.datasets as dset
import torchvision.transforms as transforms
print('\n\nCifar 10')
a = dset.CIFAR10(root="abc/def/ghi", download=True)
print(a[3])
# print('\n\nCifar 100')
# a = dset.CIFAR100(root="abc/def/ghi", download=True)
# print(a[3])
dataset = dset.CIFAR10(root='cifar', download=True, transform=transforms.ToTensor())
dataloader = torch.utils.data.DataLoader(dataset, batch_size=1,
shuffle=True, num_workers=2)
for i, data in enumerate(dataloader, 0):
print(data)
if i == 10:
break
# miter = dataloader.__iter__()
# def getBatch():
# global miter
# try:
# return miter.next()
# except StopIteration:
# miter = dataloader.__iter__()
# return miter.next()
# i=0
# while True:
# print(i)
# img, target = getBatch()
# i+=1
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment