VQ-Diffusion
VQ-Diffusion copied to clipboard
any training scripts for the pretraining the CC datasets?
Also, I want to know the training cost when training on the larger dataset like LAION. Could you tell me the training epochs when training the larger models and the relevant costage about the GPU cards and the whole training time? Thanks a lot!
In VQ-Diffusion, the largest dataset is LAION-person, which contains about 42M images, we trained it for 2 epochs. In improved VQ-Diffusion, the largest dataset is ITHQ dataset, which contains about 200M images, we also trained it for about 2 epochs on our large model, it cost about 96 V100s for a month.