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

2
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 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()