From 50ad01f52569bad19bbadda0dbc360cd951a53c3 Mon Sep 17 00:00:00 2001 From: RunningLeon Date: Tue, 5 Jul 2022 17:15:15 +0800 Subject: [PATCH] fix mmdet3d --- mmdeploy/codebase/mmdet3d/models/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmdeploy/codebase/mmdet3d/models/base.py b/mmdeploy/codebase/mmdet3d/models/base.py index e8d7000e47..61fddae8e5 100644 --- a/mmdeploy/codebase/mmdet3d/models/base.py +++ b/mmdeploy/codebase/mmdet3d/models/base.py @@ -18,6 +18,6 @@ def base3ddetector__forward_test(ctx, @FUNCTION_REWRITER.register_rewriter( 'mmdet3d.models.detectors.base.Base3DDetector.forward') -def base3ddetector__forward(ctx, self, *args): +def base3ddetector__forward(ctx, self, *args, **kwargs): """Rewrite this function to run the model directly.""" return self.forward_test(*args)