jychoi118 / P2-weighting

CVPR 2022
MIT License
143 stars 14 forks source link

train on FFHQ 512 #12

Closed Harper714 closed 1 year ago

Harper714 commented 1 year ago

Have you tried to train the model on FFHQ 512x512? Are there any suggested parameter settings?

jychoi118 commented 1 year ago

We did not try 512x512 due to heavy computational cost. You may try training diffusion model in the latent space of Stable Diffusion's VAE