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

The setting of batchsize and training time

Open pushi opened this issue 2 years ago • 1 comments

Thank you for your excellent work! We are tried to reproduce your results, but we note that the batchsize for a single GPU is very small (i.e., 4). Could you explain why you use so small batchsize on an A100 GPU? image Was the Stable diffusion model trained for about 24 days on 256 A100 GPUs? image

Could you provide the configuration file and related codes (including the dataloader and the 10% dropping of the text-condition) for training the model? we note that there is only an inference configuration file here. image

pushi avatar Sep 07 '22 12:09 pushi

@pushi did you find the answers to this question? would like to know as well

mia01 avatar Oct 21 '23 22:10 mia01