fix the issue PR pytorch#799 left (pytorch#781)

YYStreet · web-flow · commit a4b0aba292e0 · 2020-05-28T23:14:16.000-04:00
diff --git a/regression/main.py b/regression/main.py
@@ -57,7 +57,7 @@ def get_batch(batch_size=32):
 
     # Apply gradients
     for param in fc.parameters():
-        param.add_(-0.1 * param.grad)
+        param.data.add_(-0.1 * param.grad)
 
     # Stop criterion
     if loss < 1e-3:
