Skip to content

Commit ded06fe

Browse files
authored
[Fix] Fix get_flops scrip (open-mmlab#3551)
## Motivation `inputs` and `input_shape` can't be both set to mmengine api `get_model_complexity_info` open-mmlab/mmengine#1056 ## Modification Set `input_shape` to None.
1 parent 36cde54 commit ded06fe

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/analysis_tools/get_flops.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def inference(args: argparse.Namespace, logger: MMLogger) -> dict:
8686
'supported yet.')
8787
outputs = get_model_complexity_info(
8888
model,
89-
input_shape,
89+
input_shape=None,
9090
inputs=data['inputs'],
9191
show_table=False,
9292
show_arch=False)

0 commit comments

Comments
 (0)