IQA-optimization icon indicating copy to clipboard operation
IQA-optimization copied to clipboard
verified publisher icon dingkeyan93

Fix running with pytorch-cuda.

Open veluca93 opened this issue 5 years ago • 0 comments

Hello! I was attempting to run your code, and I had some issues with devices (cuda vs cpu). I think that this patch fixes the problems (it also runs some autoformatting).

veluca93 avatar Oct 07 '20 16:10 veluca93