-
Notifications
You must be signed in to change notification settings - Fork 47
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Release checkpoint for Celeb-HQ dataset #12
Comments
To increase the resolution to be x2, you need 4x GPU memory if you keep the batch size and all model architecture. From now, it is still very difficult to train an image-to-image translation in such a high resolution (i.e., 1024, as you expect). 20GB for 256 means that 320 GB (>8x Tesla v100) for 1024! Too expensive! Even more, you may need to add UNet-like Skip connection for high resolution to keep the identity while translating, the cost increases again. HiSD aims to explore the controllable diversity for broad scalable image-to-image translation area, therefore to effectively train a high-resolution model is not the main proposal but the future work. BTW, there are some papers where you may find the solutions. For example, you may utilize recent StyleGAN-based methods such as InterFaceGAN and some fast image-to-image methods such as https://arxiv.org/pdf/2012.02992.pdf. You need to combine them if you want to maintain the controllability with a high resolution. |
Thank you for your reply. Our team is amazed by the effect of HISD in the image-to-image translation area. We will continue to explore the size of 1024 according to your suggestions
At 2021-05-13 16:25:23, "imlixinyang" ***@***.***> wrote:
To increase the resolution to be x2, you need 4x GPU memory if you keep the batch size and all model architecture. From now, it is still very difficult to train an image-to-image translation in such a high resolution (i.e., 1024, as you expect).
20GB for 256 means that 320 GB (>8x Tesla v100) for 1024! Too expensive! Even more, you may need to add UNet-like Skip connection for high resolution to keep the identity while translating, the cost increases again.
HiSD aims to explore the controllable diversity for broad scalable image-to-image translation area, therefore to effectively train a high-resolution model is not the main proposal but the future work.
BTW, there are some papers where you may find the solutions. For example, you may utilize recent StyleGAN-based methods such as InterFaceGAN and some fast image-to-image methods such as https://arxiv.org/pdf/2012.02992.pdf.
—
You are receiving this because you authored the thread.
Reply to this email directly, view it on GitHub, or unsubscribe.
|
Looking forward to it! |
Thank you for your open source. This code is very helpful to my knowledge of face attribute transfer. You have opened the checkpoint of 256 size, could you open the checkpoint of 1024 size for learning and use. Thank you very much.
The text was updated successfully, but these errors were encountered: