We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a7f271e commit e529128Copy full SHA for e529128
mmengine/_strategy/base.py
@@ -322,7 +322,8 @@ def compile_model(
322
Returns:
323
nn.Module: Compiled model.
324
"""
325
- if isinstance(compile, bool) and not compile:
+ if (isinstance(compile, bool) and not compile) or \
326
+ (isinstance(compile, dict) and compile.get('disable', False)):
327
return model
328
329
assert digit_version(TORCH_VERSION) >= digit_version('2.0.0'), (
0 commit comments