Mandylove1993 / CUDA-FastBEV

TensorRT deploy and PTQ/QAT tools development for FastBEV, total time only need 6.9ms!!!
MIT License
233 stars 38 forks source link

KeyError: 'FastBEV is not in the models registry' #14

Closed LiuXuan66 closed 1 year ago

LiuXuan66 commented 1 year ago

When I run python3 ptq/dump-data.py. mmcv-full-1.4.0 mmdet3d-0.15.0 torch-1.5.1 mmdet-2.25.1 Traceback (most recent call last): File "ptq/dump-data.py", line 165, in dump_tensor(args) File "ptq/dump-data.py", line 108, in dump_tensor model = init_model(args.config, args.checkpoint, device=args.device) File "/home/liu/.local/lib/python3.8/site-packages/mmdet3d/apis/inference.py", line 57, in init_model model = build_model(config.model, test_cfg=config.get('test_cfg')) File "/home/liu/.local/lib/python3.8/site-packages/mmdet3d/models/builder.py", line 83, in build_model return build_detector(cfg, train_cfg=train_cfg, test_cfg=test_cfg) File "/home/liu/.local/lib/python3.8/site-packages/mmdet3d/models/builder.py", line 56, in build_detector return DETECTORS.build( File "/home/liu/.local/lib/python3.8/site-packages/mmcv/utils/registry.py", line 212, in build return self.build_func(*args, **kwargs, registry=self) File "/home/liu/.local/lib/python3.8/site-packages/mmcv/cnn/builder.py", line 27, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "/home/liu/.local/lib/python3.8/site-packages/mmcv/utils/registry.py", line 44, in build_from_cfg raise KeyError( KeyError: 'FastBEV is not in the models registry'

Mandylove1993 commented 1 year ago

Configure the FastBEV runtime environment first

---- Replied Message ---- | From | @.> | | Date | 09/21/2023 17:55 | | To | @.> | | Cc | @.***> | | Subject | [Mandylove1993/CUDA-FastBEV] KeyError: 'FastBEV is not in the models registry' (Issue #14) |

When I run python3 ptq/dump-data.py. Traceback (most recent call last): File "ptq/dump-data.py", line 165, in dump_tensor(args) File "ptq/dump-data.py", line 108, in dump_tensor model = init_model(args.config, args.checkpoint, device=args.device) File "/home/liu/.local/lib/python3.8/site-packages/mmdet3d/apis/inference.py", line 57, in init_model model = build_model(config.model, test_cfg=config.get('test_cfg')) File "/home/liu/.local/lib/python3.8/site-packages/mmdet3d/models/builder.py", line 83, in build_model return build_detector(cfg, train_cfg=train_cfg, test_cfg=test_cfg) File "/home/liu/.local/lib/python3.8/site-packages/mmdet3d/models/builder.py", line 56, in build_detector return DETECTORS.build( File "/home/liu/.local/lib/python3.8/site-packages/mmcv/utils/registry.py", line 212, in build return self.build_func(*args, **kwargs, registry=self) File "/home/liu/.local/lib/python3.8/site-packages/mmcv/cnn/builder.py", line 27, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "/home/liu/.local/lib/python3.8/site-packages/mmcv/utils/registry.py", line 44, in build_from_cfg raise KeyError( KeyError: 'FastBEV is not in the models registry'

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you are subscribed to this thread.Message ID: @.***>

LiuXuan66 commented 1 year ago

Got it! Thanks!