Test go wrong with only one GPU?

[lyx0208]

I run the test.sh with 1 GPU, the output is a grey image with no person. But it works with 2 GPUs, is there anyway to run with 1 GPU ?

[TianxiangMa]

Thanks, it is a model-saving error. When we saved the model checkpoint (the 157_net_netG.pth that we provided), we saved the parallel model directly, so the saved model has “module." in the keys.

We have fixed it in the line 67-74 of “models/base_model.py”.

[lyx0208]

Still goes wrong after adding these lines of codes

[TianxiangMa]

Did you use a single GPU for testing? After you modify the "base_model.py", the "--gpu_ids" in "test.sh" must be single.

[lyx0208]

Yes, and it still doesn't work

[TianxiangMa]

Did you modify the line 76 in “models/base_model.py”? It should be "new_model_dict" instead of the original ”model_dict“.

[lyx0208]

Yes

[TianxiangMa]

This problem is usually caused by the inconsistency in the model structure variable names between the loaded model and the built model. You can print the networks and check it.