fix: fix space

This commit is contained in:
wql 2024-09-05 12:59:02 +08:00
parent 0cf37e5ec1
commit f23e9d417e
1 changed files with 6 additions and 6 deletions

View File

@ -1,13 +1,13 @@
#!/bin/bash #!/bin/bash
run_type = "$1" run_type="$1"
model = "$2" model="$2"
gpu_cnt = "$3" gpu_cnt="$3"
max_steps = "$4" max_steps="$4"
current_datetime=$(date +%Y%m%d%H%M%S) 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}" run_name="${run_type}_${model}_${gpu_cnt}_gpu_${max_steps}_step_${current_datetime}"
else else
@ -36,7 +36,7 @@ python prepare_yaml_file.py ${run_type} ${model} ${max_steps} ${run_name} ${outp
# echo "Start recording gpu status " # 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 \ # ASCEND_RT_VISIBLE_DEVICES=0 llamafactory-cli train ${output_dir}/${run_name}.yml \
# | tee ${output_dir}/log.txt" & # | tee ${output_dir}/log.txt" &
# train_pid=$! # train_pid=$!