Sue

icon indicating an email [email protected]

icon company @ Harbin Institute of Technology icon location Harbin Institute of Technology, Harbin,Heilongjiang,China icon location A Ph.D student with School of Computer Science and Technology, HIT,China.

Results 2 comments of Sue

Compatibility with pytorch 1.5 & CUDA 10.2

Hi, do you solve this problem?

Compatibility with pytorch 1.5 & CUDA 10.2

AT_CHECK is deprecated in torch 1.5. You can replace AT_CHECK in nms.cpp and overlap.cpp with TORCH_CHECK