Skip to content
Snippets Groups Projects
user avatar
sjiang95 authored
* Complement type hint of get_model_complexity_info()

The type of `inputs` should be one of `torch.Tensor`,
`tuple[torch.Tensor, ...]` and `None`.

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

* Update print_helper.py

---------

Signed-off-by: default avatarShengjiang QUAN <qsj287068067@126.com>
Co-authored-by: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
5da24ed1
History
Name Last commit Last update
..