Say I have my img data in the format:
x_train, y_train, x_test, y_test
How can I turn that into something pytorch can use? I tried
train = TensorDataset(x_train, y_train) and using a dataloader, but that gives me a TypeError