p04798526
/
LLaMA-Factory-310P3
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: add echo

This commit is contained in:
wql 2024-09-05 13:09:43 +08:00
parent ceb01459fe
commit 95b4b493e6
2 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prepare_yaml_file.py
View File

@ -41,7 +41,7 @@ def main():
if run_type == "lora_sft": if run_type == "lora_sft":
config['max_steps'] = max_steps config['max_steps'] = max_steps
with open(f'{output_dir}/{run_name}.yml', 'w', encoding='utf-8') as f: with open(f'{output_dir}/{run_name}.yaml', 'w', encoding='utf-8') as f:
yaml.dump(data=config, stream=f, allow_unicode=True) yaml.dump(data=config, stream=f, allow_unicode=True)
if __name__ == "__main__": if __name__ == "__main__":

9
run_once.sh
View File

@ -18,14 +18,15 @@ output_dir="./results/${run_name}"
if [ ! -d "$output_dir" ]; then if [ ! -d "$output_dir" ]; then
mkdir -p "$output_dir" mkdir -p "$output_dir"
echo "路径不存在，已创建: $output_dir" echo "output_dir created: $output_dir"
else else
echo "路径已存在: $output_dir" echo "output_dir exists: $output_dir"
fi fi
echo "${run_type} ${model} ${gpu_cnt} ${max_steps} ${run_name} ${output_dir}"
python prepare_yaml_file.py ${run_type} ${model} ${max_steps} ${run_name} ${output_dir}
# echo "${run_type} ${model} ${gpu_cnt} ${max_steps} ${run_name} ${output_dir}"
python prepare_yaml_file.py ${run_type} ${model} ${max_steps} ${run_name} ${output_dir}
echo "yaml file save to {output_dir}/{run_name}.yaml"
# export USE_MODELSCOPE_HUB=1 # export USE_MODELSCOPE_HUB=1