From ab3d68bc13cbb28f1f028d6d4b79dc0cbce29390 Mon Sep 17 00:00:00 2001 From: Christian Risi <75698846+CnF-Gris@users.noreply.github.com> Date: Sun, 12 Oct 2025 01:41:34 +0200 Subject: [PATCH] fixed patience not quitting --- Playgrounds/nanosocrates-train-experiment-2.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/Playgrounds/nanosocrates-train-experiment-2.py b/Playgrounds/nanosocrates-train-experiment-2.py index 2a070a9..1efff13 100644 --- a/Playgrounds/nanosocrates-train-experiment-2.py +++ b/Playgrounds/nanosocrates-train-experiment-2.py @@ -331,7 +331,7 @@ while current_epoch < MAX_EPOCHS: if counter > 1: patience += 1 - + if counter == 0: patience = max(0, patience - 1) @@ -359,7 +359,7 @@ while current_epoch < MAX_EPOCHS: f"\t\tavg_txt: {txt_train_avg_loss} - avg_enc: {enc_avg_train_loss} - avg_dec: {dec_avg_train_loss}\n", f"{SEPARATOR}\n", f"Validation Losses:\n", - f"\ttxt_loss: {txt_avg_loss} - masking_loss: {enc_avg_loss} - prediction: {dec_avg_loss}\n", + f"\ttxt_loss: {txt_avg_loss} - masking_loss: {enc_avg_loss} - prediction_loss: {dec_avg_loss}\n", f"{SEPARATOR}\n", ] ) @@ -374,3 +374,6 @@ while current_epoch < MAX_EPOCHS: if current_epoch % CHECKPOINT_STEPS == 0 or patience == PATIENCE: print(f"Saving model at {CHECKPOINT_PATH.as_posix()}") torch.save(NANOSOCRATES.state_dict(), CHECKPOINT_PATH) + + if patience == PATIENCE: + exit(0)