dearleiii / PIRM-2018-SISR-Challenge

Super Resolution
https://www.pirm2018.org/PIRM-SR.html
2 stars 0 forks source link

Cant load model to Gpu #15

Closed dearleiii closed 6 years ago

dearleiii commented 6 years ago

leichen@gpu-compute3$ python3 load_model_gpu.py current device: <function current_device at 0x7fe86a746c80> Traceback (most recent call last): File "load_model_gpu.py", line 28, in model1.cuda() AttributeError: 'function' object has no attribute 'cuda' leichen@gpu-compute3$

dearleiii commented 6 years ago

map_location