pass steps_per_print to tput timer (microsoft#299)

bobisai · Jul 24, 2020 · ec94341 · ec94341
1 parent eb74c3f
commit ec94341
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deepspeed/pt/deepspeed_light.py b/deepspeed/pt/deepspeed_light.py
@@ -160,6 +160,7 @@ def __init__(self,
         self.tput_timer = ThroughputTimer(
             batch_size=self.train_micro_batch_size_per_gpu(),
             num_workers=self.dp_world_size,
+            steps_per_output=self.steps_per_print(),
             monitor_memory=False)
 
         self.training_dataloader = self.deepspeed_io(