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

To confirm the curriculum for reproduction #26

Open
Xiaoming-Zhao opened this issue Oct 15, 2021 · 0 comments
Open

To confirm the curriculum for reproduction #26

Xiaoming-Zhao opened this issue Oct 15, 2021 · 0 comments

Comments

@Xiaoming-Zhao
Copy link

Xiaoming-Zhao commented Oct 15, 2021

Hi @marcoamonteiro and @ericryanchan, congrats on this great work, and thanks a lot for open-sourcing the code!

I just checked the curriculum for training on CelebA and CATS, it seems like the curriculum only provides hyperparameters for training on the resolution of 64x64.

pi-GAN/curriculums.py

Lines 86 to 87 in 0800af7

0: {'batch_size': 28 * 2, 'num_steps': 12, 'img_size': 64, 'batch_split': 2, 'gen_lr': 6e-5, 'disc_lr': 2e-4},
int(200e3): {},

However, Sec. C in the paper states the following:

π-GAN trained for 10 hours at 32×32, 10 hours at 64×64, and 36 hours at 128×128

I hypothesize the curriculum for CARLA is what has been stated in the paper and we can use that to reproduce the results. Is this correct?

Meanwhile, I guess the gen_lr in the following might be 5e-5 instead of 4e-5 (following Sec. 3.4 in the paper)?

0: {'batch_size': 30, 'num_steps': 48, 'img_size': 32, 'batch_split': 1, 'gen_lr': 4e-5, 'disc_lr': 4e-4},

Thanks a lot in advance.

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

1 participant