Log parameter movement in GD.

linqs · Aug 17, 2023 · b5856fe · b5856fe
1 parent 166629f
commit b5856fe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ore/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java b/...ore/src/main/java/org/linqs/psl/application/learning/weight/gradient/GradientDescent.java
@@ -265,8 +265,8 @@ protected void doLearn() {
             oldObjective = objective;
 
             breakGD = breakOptimization(iteration, objective, oldObjective);
-            log.trace("Iteration {} -- Weight Learning Objective: {}, Gradient Magnitude: {}, Iteration Time: {}",
-                    iteration, objective, computeGradientNorm(), (end - start));
+            log.trace("Iteration {} -- Weight Learning Objective: {}, Gradient Magnitude: {}, Parameter Movement: {}, Iteration Time: {}",
+                    iteration, objective, computeGradientNorm(), parameterMovement, (end - start));
 
             iteration++;
         }