andre/PyTorch-Tutorial
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update to torch 0.4

Browse Source
This commit is contained in:
Morvan Zhou
2018-05-10 08:48:03 +10:00
parent 68396f5972
commit c711da855a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorial-contents/305_batch_train.py
View File

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