data-efficient-gans icon indicating copy to clipboard operation
data-efficient-gans copied to clipboard

Disable path regularization and lazy regularization

Open nupurkmr9 opened this issue 3 years ago • 3 comments

Hi, In the paper, for FFHQ 1k 256x256 training with DiffAugment, it is written that path length regularization and lazy regularization is disabled. If I am not wrong, in the DiffAugment-stylegan2-pytorch repo lazy regularization and path length regularization is still there, right? Just wanted to confirm this before I start any training. :) Thanks!

nupurkmr9 avatar Mar 19 '21 23:03 nupurkmr9

Yes. You may need to change them and possibly some other hyperparameters as said in the paper, to fully reproduce our results in the TensorFlow version.

zsyzzsoft avatar Mar 20 '21 16:03 zsyzzsoft

Thanks. Can you tell what other hyperparameters need to be changed for FFHQ with config "paper256"?

nupurkmr9 avatar Mar 20 '21 18:03 nupurkmr9

mb=32, mbstd=4, lrate=0.002, and enable the mirror augmentation

zsyzzsoft avatar Mar 21 '21 06:03 zsyzzsoft