delete cuda()

Author: Mjkim88

tutorials/06 - Recurrent Neural Network/main.py

@@ -64,7 +64,7 @@ def forward(self, x):
 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 @@ def forward(self, x):
 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')