algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-06 17:44:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

delete cuda()

Browse Source
This commit is contained in:
Mjkim88
2017-03-13 21:13:55 +09:00
committed by GitHub
parent 0108b9f2ab
commit 84b2da2c20
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tutorials/06 - Recurrent Neural Network/main.py
View File

@ -64,7 +64,7 @@ optimizer = torch.optim.Adam(rnn.parameters(), lr=learning_rate)
for epoch in range(num_epochs): for epoch in range(num_epochs):
for i, (images, labels) in enumerate(train_loader): for i, (images, labels) in enumerate(train_loader):
images = Variable(images.view(-1, sequence_length, input_size)) images = Variable(images.view(-1, sequence_length, input_size))
labels = Variable(labels).cuda() labels = Variable(labels)
# Forward + Backward + Optimize # Forward + Backward + Optimize
optimizer.zero_grad() optimizer.zero_grad()
@ -90,4 +90,4 @@ for images, labels in test_loader:
print('Test Accuracy of the model on the 10000 test images: %d %%' % (100 * correct / total)) print('Test Accuracy of the model on the 10000 test images: %d %%' % (100 * correct / total))
# Save the Model # Save the Model
torch.save(rnn, 'rnn.pkl') torch.save(rnn, 'rnn.pkl')