fixed a bug in CG that prevented it from running

pysit · Apr 27, 2015 · ebc302b · ebc302b
1 parent 0eeab2e
commit ebc302b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pysit/optimization/cg.py b/pysit/optimization/cg.py
@@ -42,7 +42,7 @@ def _select_step(self, shots, current_objective_value, gradient, iteration, obje
 
         """
 
-        reset = (self.reset_length is not None) and (not np.mod(i, self.reset_length))
+        reset = (self.reset_length is not None) and (not np.mod(iteration, self.reset_length))
 
         if (self.prev_gradient is None) or reset:
             direction = -1*gradient