While trying to load a scripted model. I get this error.
model = torch::jit::load(settings.model_path);
Error Trackeback:
loading model !!!!
unknown file: Failure
C++ exception with description "
Unknown builtin op: spconv::get_indice_pairs.
Could not find any similar ops to spconv::get_indice_pairs. This op may not exist or may not be currently supported in TorchScript.
:
File "/home/ptripathi/centernet-env/lib/python3.6/site-packages/spconv/ops.py", line 79
out_shape = spatial_shape
if grid is None:
res = torch.ops.spconv.get_indice_pairs(indices, batch_size, out_shape,
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
spatial_shape, ksize, stride,
padding, dilation, out_padding,
Serialized File "code/__torch__/spconv/ops.py", line 37
out_shape = spatial_shape
if torch.__is__(grid, None):
res = ops.spconv.get_indice_pairs(indices, batch_size, out_shape, spatial_shape, ksize, stride, padding, dilation, out_padding, int(subm), int(transpose), int(use_hash))
~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
_7 = res
else:
'get_indice_pairs' is being compiled since it was called from 'SubMConv3d.forward'
Serialized File "code/__torch__/spconv/conv.py", line 26
def forward(self: __torch__.spconv.conv.SubMConv3d,
input: __torch__.spconv.spconvtensor.SparseConvTensor) -> __torch__.spconv.spconvtensor.SparseConvTensor:
_0 = __torch__.spconv.ops.get_indice_pairs
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE
_1 = __torch__.spconv.functional.SubMConvFunction
input0 = unchecked_cast(__torch__.spconv.spconvtensor.SparseConvTensor, input)
" thrown in the test body.
Versions
Collecting environment information...
PyTorch version: 1.10.0+cu111
Is debug build: False
CUDA used to build PyTorch: 11.1
ROCM used to build PyTorch: N/A
OS: Ubuntu 18.04.6 LTS (x86_64)
GCC version: (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
Clang version: Could not collect
CMake version: version 3.13.3
Libc version: glibc-2.25
Python version: 3.6.9 (default, Jan 26 2021, 15:33:00) [GCC 8.4.0] (64-bit runtime)
Python platform: Linux-5.4.0-91-generic-x86_64-with-Ubuntu-18.04-bionic
Is CUDA available: True
CUDA runtime version: 11.1.105
GPU models and configuration: GPU 0: Quadro P2000
Nvidia driver version: 470.82.01
cuDNN version: Probably one of the following:
/usr/lib/x86_64-linux-gnu/libcudnn.so.8.0.5
/usr/lib/x86_64-linux-gnu/libcudnn_adv_infer.so.8.0.5
/usr/lib/x86_64-linux-gnu/libcudnn_adv_train.so.8.0.5
/usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8.0.5
/usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8.0.5
/usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8.0.5
/usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8.0.5
HIP runtime version: N/A
MIOpen runtime version: N/A
Versions of relevant libraries:
[pip3] numpy==1.19.5
[pip3] torch==1.10.0+cu111
[pip3] torchaudio==0.10.0+cu111
[pip3] torchvision==0.11.0+cu111
[conda] Could not collect
🐛 the bug
While trying to load a scripted model. I get this error.
Error Trackeback:
Versions
Collecting environment information... PyTorch version: 1.10.0+cu111 Is debug build: False CUDA used to build PyTorch: 11.1 ROCM used to build PyTorch: N/A
OS: Ubuntu 18.04.6 LTS (x86_64) GCC version: (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0 Clang version: Could not collect CMake version: version 3.13.3 Libc version: glibc-2.25
Python version: 3.6.9 (default, Jan 26 2021, 15:33:00) [GCC 8.4.0] (64-bit runtime) Python platform: Linux-5.4.0-91-generic-x86_64-with-Ubuntu-18.04-bionic Is CUDA available: True CUDA runtime version: 11.1.105 GPU models and configuration: GPU 0: Quadro P2000 Nvidia driver version: 470.82.01 cuDNN version: Probably one of the following: /usr/lib/x86_64-linux-gnu/libcudnn.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_adv_infer.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_adv_train.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_cnn_infer.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_cnn_train.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_ops_infer.so.8.0.5 /usr/lib/x86_64-linux-gnu/libcudnn_ops_train.so.8.0.5 HIP runtime version: N/A MIOpen runtime version: N/A
Versions of relevant libraries: [pip3] numpy==1.19.5 [pip3] torch==1.10.0+cu111 [pip3] torchaudio==0.10.0+cu111 [pip3] torchvision==0.11.0+cu111 [conda] Could not collect