wx44wx/icassp2027-experiment
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix final LoRA checkpoint report

Browse Source
This commit is contained in:
Xin Wang
2026-08-25 09:45:47 +08:00
parent aa4278e13e
commit 56c8fe0a0e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train_omni_turn_lora.py
View File

@@ -1616,7 +1616,7 @@ def _run_training(args: argparse.Namespace, run: DistributedRun) -> None:
if args.freeze_thinker: if args.freeze_thinker:
print("Frozen Thinker: no LoRA adapter was created") print("Frozen Thinker: no LoRA adapter was created")
else: else:
print(f"Saved LoRA adapter to {adapter_dir}") print(f"Saved LoRA adapter to {output_dir / 'adapter'}")
print(f"Saved turn head to {output_dir / 'turn_head.pt'}") print(f"Saved turn head to {output_dir / 'turn_head.pt'}")
print( print(
"Peak CUDA allocated across ranks: " "Peak CUDA allocated across ranks: "