From c711da855a28319bc9a839f1e169a386022e3e52 Mon Sep 17 00:00:00 2001 From: Morvan Zhou Date: Thu, 10 May 2018 08:48:03 +1000 Subject: [PATCH] update to torch 0.4 --- tutorial-contents/305_batch_train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tutorial-contents/305_batch_train.py b/tutorial-contents/305_batch_train.py index 7e58e87..06350b5 100644 --- a/tutorial-contents/305_batch_train.py +++ b/tutorial-contents/305_batch_train.py @@ -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