forward() missing 2 required positional arguments: 'x' and 's'

[panghongwei17]

when I run the python main.py --mode sample --num_domains 3 --resume_iter 100000 --w_hpf 0
--checkpoint_dir expr/checkpoints/afhq
--result_dir expr/results/afhq
--src_dir assets/representative/afhq/src
--ref_dir assets/representative/afhq/ref
then error:
forward() missing 2 required positional arguments: 'x' and 's'

[MLandCV]

If you are using multiple GPUs, you may specify just one of them using CUDA_VISIBLE_DEVICES=0 or 1 or whichever.

[panghongwei17]

Thank you. I'll try it your way may you be happy ！  

…

------------------ Original ------------------ From: mlandcv ***@***.***> Date: Fri,Sep 10,2021 1:30 PM To: clovaai/stargan-v2 ***@***.***> Cc: panghongwei17 ***@***.***>, Author ***@***.***> Subject: Re: [clovaai/stargan-v2] forward() missing 2 required positional arguments: 'x' and 's' (#114) If you are using multiple GPUs, you may specify just one of them using CUDA_VISIBLE_DEVICES=0 or 1 or whichever. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe. Triage notifications on the go with GitHub Mobile for iOS or Android.

[panghongwei17]

If you are using multiple GPUs, you may specify just one of them using CUDA_VISIBLE_DEVICES=0 or 1 or whichever.
Thank you very much for your method, the problem has been solved, wish you happiness

[Ali Waqas]

os.environ['CUDA_VISIBLE_DEVICES'] = '0'

[Khin Cho Win]

I add torch.nn.DataParallel like this:
net = torch.nn.DataParallel(model)
output = net(input)

It worked.