update to torch 0.4
This commit is contained in:
@ -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
|
||||
|
||||
Reference in New Issue
Block a user