From e1669b5a4cef23765cbed262a4786f8c6d696b5d Mon Sep 17 00:00:00 2001
From: Zchen <161216199+ZH-CEN@users.noreply.github.com>
Date: Mon, 20 Oct 2025 00:21:33 +0800
Subject: [PATCH] Increase batch size from 256 to 512 for training in
 rnn_args.yaml

---
 model_training_nnn_tpu/rnn_args.yaml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/model_training_nnn_tpu/rnn_args.yaml b/model_training_nnn_tpu/rnn_args.yaml
index 4977a59..49b86e7 100644
--- a/model_training_nnn_tpu/rnn_args.yaml
+++ b/model_training_nnn_tpu/rnn_args.yaml
@@ -74,7 +74,7 @@ dataset:
     smooth_kernel_std: 2 # standard deviation of the smoothing kernel applied to the data
 
   neural_dim: 512 # dimensionality of the neural data
-  batch_size: 256 # batch size for training (reduced for TPU memory constraints)
+  batch_size: 512 # batch size for training (reduced for TPU memory constraints)
   n_classes: 41 # number of classes (phonemes) in the dataset
   max_seq_elements: 500 # maximum number of sequence elements (phonemes) for any trial
   days_per_batch: 4 # number of randomly-selected days to include in each batch