search

genforce / genforce

An efficient PyTorch library for deep generative modeling.
https://genforce.github.io/
MIT License
907 stars 75 forks source link

Is StyleGAN2 code provided in this repo fully tested? #7

Open lzhbrian opened 3 years ago

lzhbrian commented 3 years ago

Hi, thanks for the repo! Is StyleGAN2 code provided in this repo fully tested? https://github.com/genforce/genforce/blob/master/models/stylegan2_generator.py https://github.com/genforce/genforce/blob/master/models/stylegan2_discriminator.py Thanks.

justimyhxu commented 3 years ago

We have devoted ourselves to CVPR. We will release the stylegan2 training script recently.

lzhbrian commented 3 years ago

Thanks, good luck on your CVPR submission!

On Wed, Nov 25, 2020 at 22:18 justimyhxu notifications@github.com wrote:

We have devoted ourselves to CVPR. We will release the stylegan2 training script recently.

— You are receiving this because you authored the thread.

Reply to this email directly, view it on GitHub https://github.com/genforce/genforce/issues/7#issuecomment-733733981, or unsubscribe https://github.com/notifications/unsubscribe-auth/ADBN32ALSIK3AGJA3RFBKTTSRUGZ3ANCNFSM4UAQATSQ .

hdjsjyl commented 3 years ago

Hi Author, Just want to ask whether the StyleGAN2 code provided can be used for the pre-trained model in Model_zoo. Could I only use the code for inference? Looking forward to the full release of StyleGAN2. Thank you!

justimyhxu commented 3 years ago

Now Genforce supports the inference for StyleGAN2. The training code will be released soon.

swamiviv commented 3 years ago

@justimyhxu Big fan of the repo. Any update here re StyleGAN2 training code? Thanks.

zhenglisec commented 3 years ago

Also waiting for the stylegan2 training script !!! and have you trained stylegan2 on ffhq256? could you provide the pre-trained model? Thanks.

swyoon commented 2 years ago

Thanks for the super cool repo. The training script for StyleGAN2 should be awesome. Also waiting for it!

mxl1990 commented 2 years ago

I try to train a StyleGAN2 using your scripts, but find following error: Traceback (most recent call last): File "./train.py", line 123, in <module> main() File "./train.py", line 119, in main runner.train() File "/home/lixiao/stylegan2-reid/runners/base_runner.py", line 300, in train self.train_step(data_batch, **train_kwargs) File "/home/lixiao/stylegan2-reid/runners/stylegan_runner.py", line 34, in train_step G.synthesis.lod.data.fill_(self.lod) File "/home/lixiao/anaconda3/envs/pytorch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1178, in __getattr__ type(self).__name__, name)) AttributeError: 'SynthesisModule' object has no attribute 'lod' So I am wondering that you update your scripts such as stylegan2_generator。