diff --git a/model_training_nnn/rnn_args.yaml b/model_training_nnn/rnn_args.yaml index 2944d29..81a2957 100644 --- a/model_training_nnn/rnn_args.yaml +++ b/model_training_nnn/rnn_args.yaml @@ -70,7 +70,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: 64 # batch size for training + batch_size: 32 # batch size for training 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