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-12 20:34:07 +08:00
parent 1985386c5f
commit bc015f5efb
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/rnn_trainer.py
View File

@@ -46,10 +46,10 @@ class BrainToTextDecoder_Trainer:
gradient_accumulation_steps=args.get('gradient_accumulation_steps', 1), gradient_accumulation_steps=args.get('gradient_accumulation_steps', 1),
log_with=None, # We'll use our own logging log_with=None, # We'll use our own logging
project_dir=args.get('output_dir', './output'), project_dir=args.get('output_dir', './output'),
even_batches=False, # Required for batch_size=None DataLoaders
) )
# Note: even_batches is handled automatically by Accelerator based on our DataLoader configuration # Set even_batches to False after initialization - required for batch_size=None DataLoaders
self.accelerator.even_batches = False
# Trainer fields # Trainer fields
self.args = args self.args = args