From 1929c25581ab4a077772bcf18e1c76997e46c9d4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=B0=A2=E6=98=95=E8=BE=B0?= Date: Wed, 31 Jan 2024 19:05:54 +0800 Subject: [PATCH] [Fix] Fix get_flops scrip (#3551) ## Motivation `inputs` and `input_shape` can't be both set to mmengine api `get_model_complexity_info` https://github.com/open-mmlab/mmengine/pull/1056 ## Modification Set `input_shape` to None. --- tools/analysis_tools/get_flops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/analysis_tools/get_flops.py b/tools/analysis_tools/get_flops.py index 66b2d52fcd..78a73988d4 100644 --- a/tools/analysis_tools/get_flops.py +++ b/tools/analysis_tools/get_flops.py @@ -86,7 +86,7 @@ def inference(args: argparse.Namespace, logger: MMLogger) -> dict: 'supported yet.') outputs = get_model_complexity_info( model, - input_shape, + input_shape=None, inputs=data['inputs'], show_table=False, show_arch=False)