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'), (