delete cuda()

tutorials/06 - Recurrent Neural Network/main.py

@@ -64,7 +64,7 @@ optimizer = torch.optim.Adam(rnn.parameters(), lr=learning_rate)
 for epoch in range(num_epochs):
     for i, (images, labels) in enumerate(train_loader):
         images = Variable(images.view(-1, sequence_length, input_size))
-        labels = Variable(labels).cuda()
+        labels = Variable(labels)
         
         # Forward + Backward + Optimize
         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)) 
 
 # Save the Model
-torch.save(rnn, 'rnn.pkl')
+torch.save(rnn, 'rnn.pkl')