Skip to content

Commit c711da8

Browse files
committed
update to torch 0.4
1 parent 68396f5 commit c711da8

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tutorial-contents/305_batch_train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
x = torch.linspace(1, 10, 10) # this is x data (torch tensor)
1717
y = torch.linspace(10, 1, 10) # this is y data (torch tensor)
1818

19-
torch_dataset = Data.TensorDataset(data_tensor=x, target_tensor=y)
19+
torch_dataset = Data.TensorDataset(x, y)
2020
loader = Data.DataLoader(
2121
dataset=torch_dataset, # torch TensorDataset format
2222
batch_size=BATCH_SIZE, # mini batch size

0 commit comments

Comments
 (0)