How long does it take to train the model on CIFAR-10

gnobitab / RectifiedFlow

Official Implementation of Rectified Flow (ICLR2023 Spotlight)

663 stars 40 forks source link

Closed forever208 closed 7 months ago

forever208 commented 8 months ago

The default iteration is 1300k, which gonna take 5 days using a single A100... Does the training really need 5 days?

forever208 commented 8 months ago

@gnobitab Hi, it would be appreciated if you could share your training time on A100

yuanzhi-zhu commented 7 months ago

To the best of my knowledge, it takes ~3 days on a 3090 GPU

forever208 commented 7 months ago

@yuanzhi-zhu thanks!