修复batch_size校验错误

model_training_nnn/rnn_trainer.py

@@ -48,6 +48,9 @@ class BrainToTextDecoder_Trainer:
             project_dir=args.get('output_dir', './output'),
         )
 
+        # Set even_batches to False to handle batch_size=None in DataLoaders
+        self.accelerator.even_batches = False
+
         # Trainer fields
         self.args = args
         self.logger = None