update to torch 0.4

This commit is contained in:
Morvan Zhou
2018-05-10 08:48:03 +10:00
parent 68396f5972
commit c711da855a

View File

@ -16,7 +16,7 @@ BATCH_SIZE = 5
x = torch.linspace(1, 10, 10) # this is x data (torch tensor)
y = torch.linspace(10, 1, 10) # this is y data (torch tensor)
torch_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y)
torch_dataset = Data.TensorDataset(x, y)
loader = Data.DataLoader(
dataset=torch_dataset, # torch TensorDataset format
batch_size=BATCH_SIZE, # mini batch size