gnobitab / RectifiedFlow

Official Implementation of Rectified Flow (ICLR2023 Spotlight)
663 stars 40 forks source link

How long does it take to train on LSUN bedroom? #20

Open aiihn opened 2 months ago

aiihn commented 2 months ago

Thanks for your great work! If I want to train on LSUN bedroom and get a LSUN bedroom checkpoint like in RectifiedFlow pre-trained-checkpoints, how long will it take?

I am training on 8A100, and from step100 to step1400, it takes 22mins. And in configs/default_lsun_configs.py, training.n_iters = 2400001. So it will take about 690 hours (29 days) in total. Is 29 days on 8xA100 a similar training cost time to yours?