pass steps_per_print to tput timer (#299)

ec943410 · Jeff Rasley · GitHub · eb74c3f1 · ec943410
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

deepspeed/pt/deepspeed_light.py deepspeed/pt/deepspeed_light.py +1 -0

未找到文件。
--- a/deepspeed/pt/deepspeed_light.py
+++ b/deepspeed/pt/deepspeed_light.py
@@ -160,6 +160,7 @@ class DeepSpeedLight(Module):
        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(