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

sim1st · May 29, 2020 · a4b0aba · a4b0aba
1 parent 984700d
commit a4b0aba
Showing 1 changed file with 1 addition and 1 deletion.
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: