Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "ce37a9349aa72ce15c18870e3ab73a96bec61aa5"
Unverified Commit ec943410 authored by Jeff Rasley's avatar Jeff Rasley Committed by GitHub
Browse files

pass steps_per_print to tput timer (#299)

parent eb74c3f1
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
deepspeed/pt/deepspeed_light.py
View file @ ec943410
...@@ -160,6 +160,7 @@ class DeepSpeedLight(Module): ...@@ -160,6 +160,7 @@ class DeepSpeedLight(Module):
self.tput_timer = ThroughputTimer( self.tput_timer = ThroughputTimer(
batch_size=self.train_micro_batch_size_per_gpu(), batch_size=self.train_micro_batch_size_per_gpu(),
num_workers=self.dp_world_size, num_workers=self.dp_world_size,
steps_per_output=self.steps_per_print(),
monitor_memory=False) monitor_memory=False)
self.training_dataloader = self.deepspeed_io( self.training_dataloader = self.deepspeed_io(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment