fix

mosaicml · Feb 23, 2024 · ebb40df · ebb40df
1 parent 1a53568
commit ebb40df
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,