[zero] remove misleading dtype log (microsoft#2732)

Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
stas00 · Jan 26, 2023 · a60e31a · a60e31a
1 parent 30d3f5d
commit a60e31a
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/deepspeed/runtime/engine.py b/deepspeed/runtime/engine.py
@@ -1524,8 +1524,7 @@ def _configure_zero_optimizer(self, optimizer):
             round_robin_gradients = self.zero_round_robin_gradients()
             assert not isinstance(optimizer, DummyOptim), "zero stage {} requires an optimizer".format(zero_stage)
 
-            log_dist('Creating fp16 ZeRO stage {} optimizer'.format(zero_stage),
-                     ranks=[0])
+            log_dist('Creating ZeRO stage {} optimizer'.format(zero_stage), ranks=[0])
             # Overlap and contiguous grads are meaningless in stage 1 and are ignored
             if zero_stage == ZeroStageEnum.optimizer_states:
                 overlap_comm = False