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)