[Enhance] Make the parameters of get_model_complexity_info() friendly (…

…#1056)

* print_helper: optimize inputs of get_model_complexity_info

Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>

* directly throw error

When "input_shape" and "inputs" are both `None` or both set,
throw ValueError.

Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>

---------

Signed-off-by: Shengjiang QUAN <qsj287068067@126.com>

mmengine/analysis/print_helper.py

@@ -675,7 +675,7 @@ def complexity_stats_table(
 
 def get_model_complexity_info(
     model: nn.Module,
-    input_shape: tuple,
+    input_shape: tuple = None,
     inputs: Optional[torch.Tensor] = None,
     show_table: bool = True,
     show_arch: bool = True,
@@ -696,6 +696,11 @@ def get_model_complexity_info(
     Returns:
         dict: The complexity information of the model.
     """
+    if input_shape is None and inputs is None:
+        raise ValueError('One of "input_shape" and "inputs" should be set.')
+    elif input_shape is not None and inputs is not None:
+        raise ValueError('"input_shape" and "inputs" cannot be both set.')
+
     if inputs is None:
         inputs = (torch.randn(1, *input_shape), )