projector error #3
Description
Hi,
When try to project an image to latent space by running " python projector.py --outdir=./ --target=../images/xxx.png --posefile=../images/pose-annotations.csv --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada-pytorch/pretrained/ffhq.pkl",
I got an error like this:
TypeError: forward() takes 2 positional arguments but 3 were given
stack is:
Traceback (most recent call last):
File "/opt/conda/lib/python3.8/pdb.py", line 1704, in main
pdb._runscript(mainpyfile)
File "/opt/conda/lib/python3.8/pdb.py", line 1573, in _runscript
self.run(statement)
File "/opt/conda/lib/python3.8/bdb.py", line 580, in run
exec(cmd, globals, locals)
File "<string>", line 1, in <module>
File "/workspace/TryOnGAN/projector.py", line 9, in <module>
"""Project given image to the latent space of pretrained network pickle."""
File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 829, in __call__
return self.main(*args, **kwargs)
File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 782, in main
rv = self.invoke(ctx)
File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 1066, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "/opt/conda/lib/python3.8/site-packages/click/core.py", line 610, in invoke
return callback(*args, **kwargs)
File "/workspace/TryOnGAN/projector.py", line 227, in run_projection
projected_w_steps = project(
File "/workspace/TryOnGAN/projector.py", line 97, in project
synth_images = G.synthesis(ws, pose, ret_pose=False, noise_mode='const')
File "/opt/conda/lib/python3.8/site-packages/torch/nn/modules/module.py", line 744, in _call_impl
result = self.forward(*input, **kwargs)
TypeError: forward() takes 2 positional arguments but 3 were given
Any idea what is going wrong?
Thanks.