pass steps_per_print to tput timer (#299)

ec943410 · Jeff Rasley · GitHub · eb74c3f1 · ec943410
Unverified Commit ec943410 authored Jul 23, 2020 by Jeff Rasley Committed by GitHub Jul 23, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

No files found.
--- 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(