Set patiance to 10

sergiuionescu · Dec 20, 2023 · 9312710 · 9312710
1 parent 2224849
commit 9312710
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sym/js/actor.critic.js b/sym/js/actor.critic.js
@@ -224,7 +224,7 @@ class ActorCritic {
       batchSize: batchSize,
       callbacks: [
         new tf.CustomCallback({onEpochEnd: onActorEpochEnd}),
-        tf.callbacks.earlyStopping({monitor: 'loss', verbose: 2})
+        tf.callbacks.earlyStopping({monitor: 'loss', verbose: 2, patience: 10})
       ]
     }).then(info => {
         this.latestActorLoss = info.history.loss.slice(-1);
@@ -237,7 +237,7 @@ class ActorCritic {
       batchSize: batchSize,
       callbacks: [
         new tf.CustomCallback({onEpochEnd: onCriticEpochEnd}),
-        tf.callbacks.earlyStopping({monitor: 'loss', verbose: 2})
+        tf.callbacks.earlyStopping({monitor: 'loss', verbose: 2, patience: 10})
       ]
     }).then(info => {
         this.latestCriticLoss = info.history.loss.slice(-1);