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

fix: fix max steps

This commit is contained in:
wql 2024-09-05 15:54:33 +08:00
parent f15e37dfad
commit 113966157c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
prepare_yaml_file.py
View File

@ -43,7 +43,7 @@ def main():
config['template'] = template config['template'] = template
config['output_dir'] = output_dir config['output_dir'] = output_dir
if run_type == "lora_sft": if run_type == "lora_sft":
config['max_steps'] = max_steps config['max_steps'] = int(max_steps)
with open(f'{output_dir}/{run_name}.yaml', '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)