Compatibility with pytorch 1.5 & CUDA 10.2

[jmarsil]

Just wondering if work has been done to update the scripts to support the newest versions of PyTorch and CUDA? I got the following error when trying to install the custom module for bounding box NMS and overlap calculation. I assume this is a required package for bounding box calculation applied in brain_reader.py?

Cuda Version: 10.2
Pytorch Version: 1.5

Any help would be greatly appreciated!

[Suiiyu]

Hi, do you solve this problem?

[Suiiyu]

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