lijiankuan
4/9/2019 - 8:49 AM

[PyTorch Dataset API] #pytorch

[PyTorch Dataset API] #pytorch

Dataset API

def __init__(self, loader=default_loader):
    pass
def __getitem__(self, index):
    pass
def __len__(self):
    pass

train_data  = trainset()
trainloader = DataLoader(train_data, batch_size=4,shuffle=True)