From f23e9d417ee1cfcb9c2383c3e14cae7fa39cb5df Mon Sep 17 00:00:00 2001 From: wql Date: Thu, 5 Sep 2024 12:59:02 +0800 Subject: [PATCH] fix: fix space --- run_once.sh | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/run_once.sh b/run_once.sh index 09786827..7badc3fe 100644 --- a/run_once.sh +++ b/run_once.sh @@ -1,13 +1,13 @@ #!/bin/bash -run_type = "$1" -model = "$2" -gpu_cnt = "$3" -max_steps = "$4" +run_type="$1" +model="$2" +gpu_cnt="$3" +max_steps="$4" current_datetime=$(date +%Y%m%d%H%M%S) -if [ "${run_type}" = "lora_sft" ]; then +if [ "${run_type}"="lora_sft" ]; then run_name="${run_type}_${model}_${gpu_cnt}_gpu_${max_steps}_step_${current_datetime}" else @@ -36,7 +36,7 @@ python prepare_yaml_file.py ${run_type} ${model} ${max_steps} ${run_name} ${outp # echo "Start recording gpu status " -# if [ "${gpu_cnt}" = "1" ]; then +# if [ "${gpu_cnt}"="1" ]; then # ASCEND_RT_VISIBLE_DEVICES=0 llamafactory-cli train ${output_dir}/${run_name}.yml \ # | tee ${output_dir}/log.txt" & # train_pid=$!