Fix logistic regression example

emay201 · Jan 18, 2017 · cad2233 · cad2233
1 parent 182f862
commit cad2233
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/logreg/main.py b/logreg/main.py
@@ -54,7 +54,7 @@ def get_batch(batch_size=32):
 
         # Reset gradients
         for param in fc.parameters():
-            param.grad.zero_()
+            param.grad.data.zero_()
 
         # Forward pass
         output = l1(fc(batch_x), batch_y)
@@ -65,7 +65,7 @@ def get_batch(batch_size=32):
 
         # Apply gradients
         for param in fc.parameters():
-            param.data.add_(-1 * param.grad)
+            param.data.add_(-1 * param.grad.data)
 
         # Stop criterion
         if loss < 0.1: