Update GradientDescent.java

linqs · Feb 2, 2024 · ab41d57 · ab41d57
1 parent e2bd4a2
commit ab41d57
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
@@ -335,12 +335,12 @@ protected void doLearn() {
 
             if (log.isTraceEnabled() && (evaluation != null) && (epoch % trainingEvaluationComputePeriod == 0)) {
                 runTrainingEvaluation(epoch);
-                log.trace("MAP State Training Evaluation Metric: {}", currentTrainingEvaluationMetric);
+                log.info("Epoch: {}, MAP State Training Evaluation Metric: {}", epoch, currentTrainingEvaluationMetric);
             }
 
             if (runValidation && (epoch % validationEvaluationComputePeriod == 0)) {
                 runValidationEvaluation(epoch);
-                log.debug("Current MAP State Validation Evaluation Metric: {}", currentValidationEvaluationMetric);
+                log.info("Epoch: {}, Current MAP State Validation Evaluation Metric: {}", epoch, currentValidationEvaluationMetric);
             }
 
             if (epoch % trainingStopComputePeriod == 0) {
@@ -403,7 +403,7 @@ protected void doLearn() {
             setFullModel();
 
             epoch++;
-            log.trace("Epoch: {}, Weight Learning Objective: {}, Iteration Time: {}", epoch, averageBatchObjective, (end - start));
+            log.info("Epoch: {}, Weight Learning Objective: {}, Iteration Time: {}", epoch, averageBatchObjective, (end - start));
         }
         log.info("Gradient Descent Weight Learning Finished.");