Mmdetection: supports for torch 1.5?

Created on 9 Apr 2020  Â·  2Comments  Â·  Source: open-mmlab/mmdetection

torch 1.5 build (nightly) got error :

mmdet/ops/nms/src/nms_cuda.cpp:4:23: error: ‘AT_CHECK’ was not declared in this scope
 #define CHECK_CUDA(x) AT_CHECK(x.type().is_cuda(), #x, " must be a CUDAtensor ")

Most helpful comment

Thanks for your bug report. For now, we mainly support pytorch 1.1-1.4. We will support Pytorch 1.5 in the future ASAP.

All 2 comments

Thanks for your bug report. For now, we mainly support pytorch 1.1-1.4. We will support Pytorch 1.5 in the future ASAP.

2513

Was this page helpful?
0 / 5 - 0 ratings