Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[Enhance] Complement type hint of get_model_complexity_info() (#1064)
* 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: Shengjiang QUAN <qsj287068067@126.com> * Update print_helper.py --------- Signed-off-by: Shengjiang QUAN <qsj287068067@126.com> Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
- Loading branch information