Do not divide by 0 for deep atom value movement.

linqs · Sep 12, 2023 · 92306a3 · 92306a3
1 parent 3028ff9
commit 92306a3
Showing 1 changed file with 3 additions and 1 deletion.
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
@@ -472,7 +472,9 @@ protected void measureEpochParameterMovement() {
                 deepAtomValueMovement += Math.pow(epochStartDeepAtomValues[i] - trainFullTermStore.getAtomStore().getAtomValues()[i], 2.0f);
             }
         }
-        deepAtomValueMovement = deepAtomValueMovement / numDeepAtoms;
+        if (numDeepAtoms > 0) {
+            deepAtomValueMovement = deepAtomValueMovement / numDeepAtoms;
+        }
         log.trace("Epoch Deep Atom Value Movement: {}", deepAtomValueMovement);
 
         // By default, there are no internal parameters.