From 197a976dfd2c058f1ea24ede492a401cf475d73f Mon Sep 17 00:00:00 2001 From: Oliver Date: Mon, 28 Apr 2025 23:33:21 +0100 Subject: [PATCH] cfg --- .../external_generalisation/math_curriculum_qwen_7b.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml b/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml index 6c02b61e..3a35d28b 100644 --- a/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml +++ b/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml @@ -64,8 +64,8 @@ data: prompt_key: prompt max_prompt_length: 4096 max_response_length: 2048 - train_batch_size: 8 - val_batch_size: 8 + train_batch_size: 4 + val_batch_size: 4 return_raw_chat: True return_raw_input_ids: True