From ebb40df483189bb94e13b34e84cac892c55ecfa3 Mon Sep 17 00:00:00 2001 From: Your Name Date: Fri, 23 Feb 2024 15:59:29 +0000 Subject: [PATCH] fix --- composer/trainer/mosaic_fsdp_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/composer/trainer/mosaic_fsdp_utils.py b/composer/trainer/mosaic_fsdp_utils.py index 1da6811b09..3f35df48a6 100644 --- a/composer/trainer/mosaic_fsdp_utils.py +++ b/composer/trainer/mosaic_fsdp_utils.py @@ -679,7 +679,7 @@ def _init_core_state_t2p3p0( state.training_state = TrainingState.IDLE state._is_root = None state._free_event_queue = _FreeEventQueue() - state._debug_level = dist.get_debug_level() + state._debug_level = torch.distributed.get_debug_level() state._exec_order_data = exec_order_utils._ExecOrderData( state._debug_level, backward_prefetch_limit,