From d4e19056ea548cf78135064680186969bd974842 Mon Sep 17 00:00:00 2001 From: Oliver Date: Mon, 28 Apr 2025 21:27:00 +0100 Subject: [PATCH] cfg --- .../external_generalisation/math_curriculum_qwen_7b.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml b/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml index 2fe8c2e0..f313bccc 100644 --- a/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml +++ b/training/configs/external_generalisation/math_curriculum_qwen_7b.yaml @@ -171,7 +171,7 @@ trainer: logger: [ 'console', 'wandb' ] val_generations_to_log_to_wandb: 0 nnodes: 1 - n_gpus_per_node: 2 + n_gpus_per_node: 4 save_freq: 100 # auto: find the last ckpt to resume. If can't find, start from scratch resume_mode: auto # or auto or resume_path if