CescMessi 83d76abc7f

[Fix] Fix the incorrect device of inputs in get_model_complexity_info (#1130)

2023-05-06 10:55:17 +08:00

..

__init__.py

[Feature] Support model complexity computation (#779)

2023-02-20 15:00:28 +08:00

complexity_analysis.py

Refine the style of table (#941)

2023-02-20 20:12:37 +08:00

jit_analysis.py

[Feature] Support model complexity computation (#779)

2023-02-20 15:00:28 +08:00

jit_handles.py

[Fix] Add PyTorch 2.0 CI and fix unit tests (#1026)

2023-03-29 13:09:23 +08:00

print_helper.py

[Fix] Fix the incorrect device of inputs in get_model_complexity_info (#1130)

2023-05-06 10:55:17 +08:00