### Something Wrong ###
Trying to evaluate your codes: WSL2 under Windows 10 Nvidia RTX3090
Upon start provided script for inference (regardless of using ckpts) , from the very beginning got a message:
### Something Wrong ###
0%|
After that calculation continue. If "--FFHQ" or "--Places2_Nature" specified - inference finished with no error.
However if " ImageNet" specified - inference finished with an error:
raise AssertionError("Invalid device id")
AssertionError: Invalid device id
NVCC report:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_19:00:59_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0
Tried to change line 44 in "run.py" from
CUDA_VISIBLE_DEVICES=0,1
to
CUDA_VISIBLE_DEVICES=0
-- no luck.
Found the reason for message:
### Something Wrong ###
0%|
That happened due to the different file image/mask names (they should be identical). However the rest of the issue still exists. Beside of that after finishing inference (regardless specified ckpts) "output" sub folder in the folder "Guided_Upsample" got created but it is empty. "output" sub folder in the folder "Transformer" got created and consists generated image with 32 or 48 pixel.
I have the same question, but try to change the file "Guided_Upsample/src/models.py" at line 72, it seems wroks: generator = nn.DataParallel(generator, device_ids=[0]) discriminator = nn.DataParallel(discriminator , device_ids=[0])