PoliBa-DeepLearning/NanoSocrates
2
0
Fork 0
Code Issues 3 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

Fixed Patience system

Browse Source
This commit is contained in:
Christian Risi 2025-10-12 01:22:06 +02:00
parent f98f5a2611
commit 79438e3d30
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Playgrounds/nanosocrates-train-experiment-2.py
View File

@ -331,6 +331,9 @@ while current_epoch < MAX_EPOCHS:
if counter > 1: if counter > 1:
patience += 1 patience += 1
if counter == 0:
patience = max(0, patience - 1)
txt_train_avg_loss = sum(text_batch_losses) / len(text_batch_losses) txt_train_avg_loss = sum(text_batch_losses) / len(text_batch_losses)