diff --git a/代码/code/infer.py b/代码/code/infer.py index 77f5250..9bca191 100644 --- a/代码/code/infer.py +++ b/代码/code/infer.py @@ -504,14 +504,6 @@ def load_model(ckpt_path, device='cuda:0'): print(f"[WARNING] Checkpoint {ckpt_path} not found, using random weights") model.to(dev) - - # === torch.compile + dynamic=True:告知编译器形状可变,避免重编译 === - try: - model = torch.compile(model, dynamic=True) - print(f"[INFO] torch.compile applied (dynamic=True)") - except Exception as e: - print(f"[WARNING] torch.compile failed ({e}), using original model") - model.eval() print(f"[INFO] Model ready. Device: {dev}")