mmengine

mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00

History

[Fix] Support calculate the flops of matmul with single dimension matrix (#970 )

* Support calculate the flops of matmul

* Remove unnecessary type ignore

* Update mmengine/analysis/jit_handles.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

2023-03-09 17:29:26 +08:00

test_activation_count.py

[Feature] Support model complexity computation (#779 )

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

test_flop_count.py

[Fix] Support calculate the flops of matmul with single dimension matrix (#970 )

2023-03-09 17:29:26 +08:00

test_jit_analysis.py

[Feature] Support model complexity computation (#779 )

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

test_param_count.py

[Feature] Support model complexity computation (#779 )

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