Fixed Patience system

2025-10-12 01:22:06 +02:00 · 2025-10-12 01:22:06 +02:00 · 79438e3d30
commit 79438e3d30
parent f98f5a2611
1 changed files with 3 additions and 0 deletions
--- a/Playgrounds/nanosocrates-train-experiment-2.py
+++ b/Playgrounds/nanosocrates-train-experiment-2.py
@ -332,6 +332,9 @@ while current_epoch < MAX_EPOCHS:
            if counter > 1:
                patience += 1
            
+            if counter == 0:
+                patience = max(0, patience - 1)
+

        txt_train_avg_loss = sum(text_batch_losses) / len(text_batch_losses)