From 9175f7c9bb7a388f5d5cc1745f058eb7399d8925 Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 8 Aug 2024 09:18:29 +0800 Subject: [PATCH] fix: change to GPUS_PER_NODE=1 --- FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh b/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh index d52e175..abbc9d2 100644 --- a/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh +++ b/FM_9G/apps/fm9g_2b/pretrain_dragonfly.sh @@ -219,7 +219,7 @@ else fi -GPUS_PER_NODE=2 +GPUS_PER_NODE=1 NNODES=1 RANK=0 MASTER_ENDPOINT=ubuntu