fix small bug in gp regression example

hjmshi · Dec 1, 2018 · 139fb18 · 139fb18
1 parent 7528c89
commit 139fb18
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/Gaussian_Processes/gp_regression.py b/examples/Gaussian_Processes/gp_regression.py
@@ -72,8 +72,9 @@ def closure():
     return loss
 
 loss = closure()
+loss.backward()
 
-training_iter = 5
+training_iter = 10
 for i in range(training_iter):
 
     # perform step and update curvature