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

17
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
) )
for epoch in range(3): # train entire dataset 3 times
for step, (batch_x, batch_y) in enumerate(loader): # for each training step def show_batch():
# train your data... for epoch in range(3): # train entire dataset 3 times
print('Epoch: ', epoch, '| Step: ', step, '| batch x: ', for step, (batch_x, batch_y) in enumerate(loader): # for each training step
batch_x.numpy(), '| batch y: ', batch_y.numpy()) # train your data...
print('Epoch: ', epoch, '| Step: ', step, '| batch x: ',
batch_x.numpy(), '| batch y: ', batch_y.numpy())
if __name__ == '__main__':
show_batch()