lucidrains/DALLE-pytorch

Deepspeed and training code print different throughput

wintersurvival opened this issue · 1 comments

When training with 8 GPU, the throughput printed by Deepspeed is much smaller than throughput calculated by training code:
deepspeed SamplesPerSec=505
sample_per_sec: 50120

It seems that the throughput calculated by training code = throughput printed by Deepspeed * gradient_steps
Which number is accurate? @lucidrains @janEbert

Hey! @rom1504 implemented that calculation. :)