Update GradientDescent.java

linqs · Aug 31, 2023 · 3f92c8d · 3f92c8d
1 parent 61d830b
commit 3f92c8d
Showing 1 changed file with 3 additions and 3 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
@@ -356,7 +356,6 @@ protected void doLearn() {
                 log.trace("Batch: {} -- Weight Learning Objective: {}, Gradient Magnitude: {}, Iteration Time: {}",
                         batchId, batchObjective, computeGradientNorm(), (batchEnd - batchStart));
             }
-            epoch++;
 
             for (DeepPredicate deepPredicate : deepPredicates) {
                 deepPredicate.epochEnd();
@@ -365,8 +364,6 @@ protected void doLearn() {
             long end = System.currentTimeMillis();
             totalTime += end - start;
 
-            log.trace("Epoch: {} -- Iteration Time: {}", epoch, (end - start));
-
             if (epoch % trainingStopComputePeriod == 0) {
                 setFullTrainModel();
 
@@ -375,6 +372,9 @@ protected void doLearn() {
 
                 breakGD = breakOptimization(epoch);
             }
+
+            epoch++;
+            log.trace("Epoch: {} -- Iteration Time: {}", epoch, (end - start));
         }
         setFullTrainModel();