ColossalAI/tests/test_fx/test_pipeline/test_timm_model
YuliangLiu0306 30b4fc0eb0
[fx]add split module pass and unit test from pipeline passes (#1242)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [fx]add split module pass and unit test from pipeline passes

* fix MNASNet bug

* polish
2022-07-12 13:45:01 +08:00
..
test_timm.py [fx]add split module pass and unit test from pipeline passes (#1242) 2022-07-12 13:45:01 +08:00
timm_utils.py [fx]add split module pass and unit test from pipeline passes (#1242) 2022-07-12 13:45:01 +08:00