ColossalAI/tests/test_fx/test_pipeline
Frank Lee 4f4d8c3656
[fx] added apex normalization to patched modules (#1300)
* [fx] added apex normalization to patched modules

* remove unused imports
2022-07-14 14:24:13 +08:00
..
test_hf_model [fx] added apex normalization to patched modules (#1300) 2022-07-14 14:24:13 +08:00
test_timm_model [fx]add split module pass and unit test from pipeline passes (#1242) 2022-07-12 13:45:01 +08:00
test_torchvision [hotfix] torchvison fx unittests miss import pytest (#1277) 2022-07-12 23:04:06 +08:00