result is black

[vbqgyy]

(PY388) G:\proj\YUs_Ai\02PYTORCH\00style_transfer\HD\00Structure-emphasized-Multimodal-Style-Transfer-master\SEMST_Original>python test.py -c ../content/avril.jpg -s ../style/horse.jpg
CUDA available: GeForce RTX 3090
SEMST_Original_model_state.pth loaded
result saved into files starting with avril_horse

————————————————————————————————————
All is good, but the result is black
win10
pytorch 1.8.1
python 3.8
rtx 3090
cuda 11.1

[irasin]

Does any error occur?

I have not tested on the newest Pytorch yet. Show me the log if possible.

[vbqgyy]

No. I tested it on the cpu and everything is good.Where is the log file? Sin ***@***.***> 于2021年5月1日周六 下午2:00写道：

…

Does any error occur? I have not tested on the newest Pytorch yet. Show me the log if possible. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#2 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AFKY26SI5EDHZKVQXDL4JDTTLOKIHANCNFSM433QQSDQ> .

[irasin]

Hi, @vbqgyy

I tested it on PyTorch1.8 cpu and it seems OK. Check your env again please.

[vbqgyy]

There is no problem with cpu, only gpu has a problem。

[irasin]

@vbqgyy

Sorry, I will check it later

[vbqgyy]

That would be great. Thank you.

[irasin]

Hi, @vbqgyy

I have tested on cuda11 + Pytorch1.8,1 and the result is wrong indeed. And the reason why the result is black, is because the output of Model is Nan.

However, when I tested it on cuda10.1 + Pytorch1.7.1, the result is OK again.

I guess there are some version compatibility problems here. I will fix it if I have time. Or you can check it by yourself. Best wish.

[vbqgyy]

@irasin
Thank you for taking the time to do the test.I am just a beginner.I guess it will take a lot of time and luck to complete this thing.