lucidrains / DALLE-pytorch

Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Deepspeed and training code print different throughput

wintersurvival opened this issue · 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. :)