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:20:15 +08:00
parent 68538746da
commit 1985386c5f
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
model_training_nnn/rnn_trainer.py
View File

@@ -46,6 +46,7 @@ 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 # Note: even_batches is handled automatically by Accelerator based on our DataLoader configuration