File tree Expand file tree Collapse file tree 1 file changed +9
-10
lines changed Expand file tree Collapse file tree 1 file changed +9
-10
lines changed Original file line number Diff line number Diff line change @@ -2212,17 +2212,16 @@ def _validate_and_set_debugger_configs(self):
2212
2212
self .debugger_hook_config = False
2213
2213
2214
2214
# Disable debugger if checkpointing is enabled by the customer
2215
- _should_disable_debugger = False
2216
2215
if self .checkpoint_s3_uri and self .checkpoint_local_path and self .debugger_hook_config :
2217
- if self .instance_count > 1 :
2218
- _should_disable_debugger = True
2219
- if hasattr (self , "distribution" ) and self .distribution is not None :
2220
- _should_disable_debugger = True
2221
- if _should_disable_debugger :
2222
- logger . info (
2223
- "SMDebug Does Not Currently Support Distributed Training Jobs With Checkpointing Enabled"
2224
- )
2225
- self .debugger_hook_config = False
2216
+ if self .instance_count > 1 \
2217
+ or (
2218
+ hasattr (self , "distribution" ) and self .distribution is not None # pylint: disable=no-member
2219
+ ):
2220
+ logger . info (
2221
+ "SMDebug Does Not Currently Support \
2222
+ Distributed Training Jobs With Checkpointing Enabled"
2223
+ )
2224
+ self .debugger_hook_config = False
2226
2225
2227
2226
def _stage_user_code_in_s3 (self ):
2228
2227
"""Upload the user training script to s3 and return the location.
You can’t perform that action at this time.
0 commit comments