zchen/b2txt25
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

tpu

Browse Source
This commit is contained in:
Zchen
2025-10-15 14:38:56 +08:00
parent e7947f310c
commit 7bdfc0d257
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
model_training_nnn_tpu/rnn_trainer.py
View File

@@ -581,11 +581,11 @@ class BrainToTextDecoder_Trainer:
val_steps_since_improvement = 0 val_steps_since_improvement = 0
# training params # training params
save_best_checkpoint = self.args.get('save_best_checkpoint', True) save_best_checkpoint = self.args.get('save_best_checkpoint', True)
early_stopping = self.args.get('early_stopping', 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() train_start_time = time.time()