improved-diffusion icon indicating copy to clipboard operation
improved-diffusion copied to clipboard

Published 20 hours ago • verified publisher icon openai

How long does training take?

Open sur-sakthy opened this issue 1 year ago • 2 comments

Hi

I am training the improved-diffusion model on the CUB dataset to obtain a .ckpt file for the DiffusionCLIP model. Does anybody know how long the training will take?

Thanks

sur-sakthy avatar Jun 07 '23 18:06 sur-sakthy

That depends on your hardware and batchsize. Per default a .pt file is generated after the 10000th step

nicolasfischoeder avatar Jun 16 '23 20:06 nicolasfischoeder

That depends on your hardware and batchsize. Per default a .pt file is generated after the 10000tthe which shape is the image shape from my dataset which is 256*256,when i train this model,how big the augument of step need to set?

1289600760 avatar Nov 19 '23 14:11 1289600760