tpu

2025-10-12 20:59:55 +08:00
parent db6108f250
commit 12d571c70b
2 changed files with 1 additions and 2 deletions
--- a/model_training_nnn/rnn_trainer.py
+++ b/model_training_nnn/rnn_trainer.py
@@ -46,7 +46,6 @@ class BrainToTextDecoder_Trainer:
            gradient_accumulation_steps=args.get('gradient_accumulation_steps', 1),
            log_with=None,  # We'll use our own logging
            project_dir=args.get('output_dir', './output'),
-            even_batches=False,  # Required for batch_size=None DataLoaders
        )