-
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:
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>
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:
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>