tpu

2025-10-15 14:38:56 +08:00
parent e7947f310c
commit 7bdfc0d257
1 changed files with 2 additions and 2 deletions
--- a/model_training_nnn_tpu/rnn_trainer.py
+++ b/model_training_nnn_tpu/rnn_trainer.py
@@ -585,7 +585,7 @@ class BrainToTextDecoder_Trainer:
        save_best_checkpoint = self.args.get('save_best_checkpoint', True)
        early_stopping = self.args.get('early_stopping', True)

-        early_stopping_val_steps = self.args['early_stopping_val_steps']
+        early_stopping_val_steps = self.args.get('early_stopping_val_steps', 20)

        train_start_time = time.time()