github
/
minimind
mirror of https://github.com/jingyaogong/minimind.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix lora save path

pull/189/head
xunuohope1107 2025-02-19 10:38:26 +08:00 committed by GitHub
parent 68762f084a
commit 45a4bb3a0f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_lora.py
View File

@ -80,7 +80,7 @@ def train_epoch(epoch, wandb):
if (step + 1) % args.save_interval == 0 and (not ddp or dist.get_rank() == 0): if (step + 1) % args.save_interval == 0 and (not ddp or dist.get_rank() == 0):
model.eval() model.eval()
# 【区别1】只保存lora权重即可 # 【区别1】只保存lora权重即可
save_lora(model, f'{args.save_dir}/lora/{args.lora_name}_{lm_config.dim}.pth') save_lora(model, f'{args.save_dir}/{args.lora_name}_{lm_config.dim}.pth')
model.train() model.train()