diff --git a/tutorials/01-basics/feedforward_neural_network/main-gpu.py b/tutorials/01-basics/feedforward_neural_network/main-gpu.py index f8b8570..3841f16 100644 --- a/tutorials/01-basics/feedforward_neural_network/main-gpu.py +++ b/tutorials/01-basics/feedforward_neural_network/main-gpu.py @@ -57,7 +57,7 @@ optimizer = torch.optim.Adam(net.parameters(), lr=learning_rate) for epoch in range(num_epochs): for i, (images, labels) in enumerate(train_loader): # Convert torch tensor to Variable - images = Variable(images.view(-1, 28*28).cuda(), requires_grad=True) + images = Variable(images.view(-1, 28*28).cuda()) labels = Variable(labels.cuda()) # Forward + Backward + Optimize