Trying to test models from github autonomous driving project

See more:

Python

machine-learning

Hello everyone, please I'm trying to test models from this Github project - GitHub - hlsheng1/CT3D: "Improving 3D Object Detection with Channel-wise Transformer", ICCV2021 accept![^] I have installed all the requirement, created the directories structures, I've downloaded the data to my Google drive and I'm working with Google Colab. When I tried to train with pre-trained models I got the Error below.

! python test.py --cfg_file ${CONFIG_FILE} --ckpt ${CKPT}

Traceback (most recent call last):
  File "test.py", line 13, in <module>
    from eval_utils import eval_utils
  File "/content/drive/My Drive/CT3D/CT3D-master/tools/eval_utils/eval_utils.py", line 8, in <module>
    from pcdet.models import load_data_to_gpu
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/__init__.py", line 6, in <module>
    from .detectors import build_detector
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/detectors/__init__.py", line 1, in <module>
    from .detector3d_template import Detector3DTemplate
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/detectors/detector3d_template.py", line 7, in <module>
    from .. import backbones_2d, backbones_3d, dense_heads, roi_heads
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/backbones_3d/__init__.py", line 2, in <module>
    from .spconv_backbone import VoxelBackBone8x, VoxelResBackBone8x
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/backbones_3d/spconv_backbone.py", line 29, in <module>
    class SparseBasicBlock(spconv.SparseModule):
AttributeError: module 'spconv' has no attribute 'SparseModule'

The solution that I found from the Issues section of this project on Github is that the latest 'spconv version' install be Google Colab is not compatible with this project, that I have to install 'spconv version 1.2.1'. Please does anyone knows the command to install 'spconv version 1.2.1' into Google Colab notebook, thanks for any help.

What I have tried:

! python test.py --cfg_file ${CONFIG_FILE} --ckpt ${CKPT}

Traceback (most recent call last):
  File "test.py", line 13, in <module>
    from eval_utils import eval_utils
  File "/content/drive/My Drive/CT3D/CT3D-master/tools/eval_utils/eval_utils.py", line 8, in <module>
    from pcdet.models import load_data_to_gpu
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/__init__.py", line 6, in <module>
    from .detectors import build_detector
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/detectors/__init__.py", line 1, in <module>
    from .detector3d_template import Detector3DTemplate
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/detectors/detector3d_template.py", line 7, in <module>
    from .. import backbones_2d, backbones_3d, dense_heads, roi_heads
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/backbones_3d/__init__.py", line 2, in <module>
    from .spconv_backbone import VoxelBackBone8x, VoxelResBackBone8x
  File "/content/drive/My Drive/CT3D/CT3D-master/pcdet/models/backbones_3d/spconv_backbone.py", line 29, in <module>
    class SparseBasicBlock(spconv.SparseModule):
AttributeError: module 'spconv' has no attribute 'SparseModule'