search

zsyOAOA / ResShift

ResShift: Efficient Diffusion Model for Image Super-resolution by Residual Shifting (NeurIPS@2023 Spotlight, TPAMI@2024)
Other
944 stars 50 forks source link

How long does it take to train on imagenet? #5

Closed yh-yoon closed 1 year ago

yh-yoon commented 1 year ago

Thank you for your awesome research. I'm trying to train with imagenet in your code, and it turns out that it takes at least 14 days to run 50k iterations. Can you tell me how long it took you to train? And is there a technique that can expedite training?

zsyOAOA commented 1 year ago

@yh-yoon Hello, it takes about 3 days with 8 x V100 GPU cards, totally 500k iterations. In fact, the model performs well after training 300k iterations.