PrimDiffusion icon indicating copy to clipboard operation
PrimDiffusion copied to clipboard
verified publisher icon FrozenBurning

Great work!! I want to know how many GPU hours do I need to train this model?

Open cuijh26 opened this issue 1 year ago • 1 comments

cuijh26 avatar May 21 '24 07:05 cuijh26

As specified in our supplementary doc, we train the denoiser on 4 A100 GPUs with a batch size of 8 on each GPU for 200,000 iterations.

FrozenBurning avatar May 22 '24 02:05 FrozenBurning