search

rosinality / stylegan2-pytorch

Implementation of Analyzing and Improving the Image Quality of StyleGAN (StyleGAN 2) in PyTorch
MIT License
2.74k stars 623 forks source link

Error in projector.py with low steps #267

Open smithee77 opened 3 years ago

smithee77 commented 3 years ago

Hi, first of all thanks for your work.

I'm having issues with projector.py when steps < 100. E.g. with !python projector.py --size 1024 --step 99 --ckpt /content/output/checkpoint/latest.pt $filepath

I get this error

Traceback (most recent call last): File "/content/stylegan_ada_ros/projector.py", line 226, in imggen, = g_ema([latent_path[-1]], input_is_latent=True, noise=noises) IndexError: list index out of range

But when I use !python projector.py --size 1024 --step 100 --ckpt /content/output/checkpoint/latest.pt $filepath everything works fine

Any clue? thanks in advance