Skip to content
Snippets Groups Projects
Unverified Commit f76218a4 authored by sjiang95's avatar sjiang95 Committed by GitHub
Browse files

[Enhance] Make the parameters of get_model_complexity_info() friendly (#1056)


* print_helper: optimize inputs of get_model_complexity_info

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

* directly throw error

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

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

---------

Signed-off-by: default avatarShengjiang QUAN <qsj287068067@126.com>
parent 5e1ed7aa
No related branches found
No related tags found
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment