From d93517e11eacd7878af2445c853fd8cbc18719fe Mon Sep 17 00:00:00 2001 From: mgam <312065559@qq.com> Date: Sat, 1 Nov 2025 15:26:28 +0800 Subject: [PATCH] Correctly process compile.disable in config --- mmengine/_strategy/base.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mmengine/_strategy/base.py b/mmengine/_strategy/base.py index 25d2ada9ec..ae1e6a4479 100644 --- a/mmengine/_strategy/base.py +++ b/mmengine/_strategy/base.py @@ -322,7 +322,8 @@ def compile_model( Returns: nn.Module: Compiled model. """ - if isinstance(compile, bool) and not compile: + if (isinstance(compile, bool) and not compile) or \ + (isinstance(compile, dict) and compile.get('disable', False)): return model assert digit_version(TORCH_VERSION) >= digit_version('2.0.0'), (