Open zywvvd opened 5 years ago
My torch environment is python 3.7, pytorch 1.2, cuda 10.0.13, cudnn 7.6.4. Under this condition, the original codes will give an error. So I fix function get_patch in common.py and function get_patch in srdata.py to solve this problem.
My torch environment is python 3.7, pytorch 1.2, cuda 10.0.13, cudnn 7.6.4. Under this condition, the original codes will give an error. So I fix function get_patch in common.py and function get_patch in srdata.py to solve this problem.