From aa3e6affa194516968913cb5a3ba6750ee009633 Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Thu, 9 Apr 2026 15:09:58 +0800 Subject: [PATCH] [update] add a comment --- trainer/train_lora.py | 1 + 1 file changed, 1 insertion(+) diff --git a/trainer/train_lora.py b/trainer/train_lora.py index 4dae568..8c51c08 100644 --- a/trainer/train_lora.py +++ b/trainer/train_lora.py @@ -96,6 +96,7 @@ if __name__ == "__main__": parser.add_argument('--from_resume', default=0, type=int, choices=[0, 1], help="是否自动检测&续训(0=否,1=是)") parser.add_argument("--use_wandb", action="store_true", help="是否使用wandb") parser.add_argument("--wandb_project", type=str, default="MiniMind-LoRA", help="wandb项目名") + # PS:多卡 DDP + LoRA 场景下建议关闭 use_compile,避免触发 torch.compile 的兼容性问题 parser.add_argument("--use_compile", default=0, type=int, choices=[0, 1], help="是否使用torch.compile加速(0=否,1=是)") args = parser.parse_args()