Skip to content
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

Closed
huang-1030 opened this issue May 13, 2021 · 3 comments
Closed

Release checkpoint for Celeb-HQ dataset #12

huang-1030 opened this issue May 13, 2021 · 3 comments

Comments

@huang-1030
Copy link

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.

@imlixinyang
Copy link
Owner

imlixinyang commented May 13, 2021

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.

@huang-1030
Copy link
Author

huang-1030 commented May 13, 2021 via email

@imlixinyang
Copy link
Owner

Looking forward to it!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants