[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)