[update] add a comment

2026-06-06 00:04:50 +00:00 · 2026-04-09 15:09:58 +08:00
parent 25a7edcd6f
commit aa3e6affa1
1 changed files with 1 additions and 0 deletions
@@ -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()