TensorRT Support in FBOCC?

[maueki]

Hi,

Does FBOCC support execution with TensorRT? When I try to run benchmark_trt.py, I encounter the following error:

$ python3 tools/analysis_tools/benchmark_trt.py ./occupancy_configs/fb_occ/fbocc-r50-cbgs_depth_16f_16x4_20e.py ckpts/fbocc-r50-cbgs_depth_16f_16x4_20e.pth
08/08 16:10:48 - mmengine - INFO - Successfully loaded tensorrt plugins from /home/blaize/work/fb-bev/fb-bev-venv/lib/python3.8/site-packages/mmdeploy/lib/libmmdeploy_tensorrt_ops.so
Traceback (most recent call last):
  File "tools/analysis_tools/benchmark_trt.py", line 190, in <module>
    fps = main()
  File "tools/analysis_tools/benchmark_trt.py", line 133, in main
    model = build_model(cfg.model, test_cfg=cfg.get('test_cfg'))
  File "/home/blaize/work/fb-bev/FB-BEV/mmdet3d/models/builder.py", line 122, in build_model
    return build_detector(cfg, train_cfg=train_cfg, test_cfg=test_cfg)
  File "/home/blaize/work/fb-bev/FB-BEV/mmdet3d/models/builder.py", line 95, in build_detector
    return MMDET_DETECTORS.build(
  File "/home/blaize/work/fb-bev/mmcv/mmcv/utils/registry.py", line 234, in build
    return self.build_func(*args, **kwargs, registry=self)
  File "/home/blaize/work/fb-bev/mmcv/mmcv/cnn/builder.py", line 27, in build_model_from_cfg
    return build_from_cfg(cfg, registry, default_args)
  File "/home/blaize/work/fb-bev/mmcv/mmcv/utils/registry.py", line 58, in build_from_cfg
    raise KeyError(
KeyError: 'FBOCCTRT is not in the models registry'

thank you.