Skip to content

GitLab

Commit 715eb7c8 authored by dcuai's avatar dcuai
Browse files

Update finetune_docowl_lora_dcu.sh HIP_VISIBLE_DEVICES

parent 78ce2481
Pipeline #1914 canceled with stages
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
finetune_docowl_lora_dcu.sh
View file @ 715eb7c8
...@@ -6,7 +6,7 @@ NODE_RANK=0 ...@@ -6,7 +6,7 @@ NODE_RANK=0
GPUS_PER_NODE=4 GPUS_PER_NODE=4
MASTER_ADDR=127.0.0.1 MASTER_ADDR=127.0.0.1
MASTER_PORT=29513 MASTER_PORT=29513
CUDA_VISIBLE_DEVICES=1,3,4,7 HIP_VISIBLE_DEVICES=1,3,4,7
# GPUS_PER_NODE=1 # GPUS_PER_NODE=1
DISTRIBUTED_ARGS="--nproc_per_node $GPUS_PER_NODE --nnodes $NNODES --node_rank $NODE_RANK --master_addr $MASTER_ADDR --master_port $MASTER_PORT" DISTRIBUTED_ARGS="--nproc_per_node $GPUS_PER_NODE --nnodes $NNODES --node_rank $NODE_RANK --master_addr $MASTER_ADDR --master_port $MASTER_PORT"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment