Hi guys,
I met a problem in using the torch.utils.data.TensorDataset and Dataloader.
According to my knowledge, I need to put the input and output in data_x and data_y,
and write code like this:
dataset = TensorDataset(data_x,data_y)
train_loader = Dataloader(dataset = dataset,batch_size = 4, shuffle = True)
But the problem is that my data_x is too large for my memory(only 8GB).
I just corp human facies from about 25K images and resize them to 224*224 to be the input of my network,
put all of these facies in data_x is not allowed by my memory.