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

fixed for windows

Browse Source
This commit is contained in:
morvanzhou
2018-07-05 16:38:07 +08:00
parent f2ebd8a37b
commit d053113d3a
1 changed files with 12 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tutorial-contents/305_batch_train.py
View File

@ -24,8 +24,15 @@ loader = Data.DataLoader(
num_workers=2, # subprocesses for loading data num_workers=2, # subprocesses for loading data
) )
def show_batch():
for epoch in range(3): # train entire dataset 3 times for epoch in range(3): # train entire dataset 3 times
for step, (batch_x, batch_y) in enumerate(loader): # for each training step for step, (batch_x, batch_y) in enumerate(loader): # for each training step
# train your data... # train your data...
print('Epoch: ', epoch, '| Step: ', step, '| batch x: ', print('Epoch: ', epoch, '| Step: ', step, '| batch x: ',
batch_x.numpy(), '| batch y: ', batch_y.numpy()) batch_x.numpy(), '| batch y: ', batch_y.numpy())
if __name__ == '__main__':
show_batch()