From f9a490daf36d6902ca3dbcb36e68ed95d435d680 Mon Sep 17 00:00:00 2001 From: susu <47966514+susu-noob@users.noreply.github.com> Date: Tue, 4 Jul 2023 16:09:40 +0800 Subject: [PATCH] fix finetune config mapping (#20) * fix finetune config mapping --- Finetune/finetune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Finetune/finetune.py b/Finetune/finetune.py index 70fe0f0be..3d6feec9b 100644 --- a/Finetune/finetune.py +++ b/Finetune/finetune.py @@ -38,7 +38,7 @@ def parse_args(): "Training.weight_decay": "optimizer.config.weight_decay", "Training.num_training_workers": [ "ray_config.scaling_config.num_workers", - "ray_config.init.runtime_env.env_vars.CCL_WORKER_COUNT#str" + "ray_config.init.runtime_env.env_vars.CCL_WORKER_COUNT#str", "ray_config.init.runtime_env.env_vars.WORLD_SIZE#str" ], "Training.resources_per_worker.CPU": [