ColossalAI/tests/test_fx/test_pipeline/test_hf_model
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
..
hf_utils.py [fx]add split module pass and unit test from pipeline passes (#1242) 2022-07-12 13:45:01 +08:00
test_albert.py [hotfix] skipped unsafe test cases (#1282) 2022-07-13 00:08:59 +08:00
test_bert.py [hotfix] skipped unsafe test cases (#1282) 2022-07-13 00:08:59 +08:00
test_gpt.py [hotfix] skipped unsafe test cases (#1282) 2022-07-13 00:08:59 +08:00
test_opt.py [hotfix] skipped unsafe test cases (#1282) 2022-07-13 00:08:59 +08:00
test_t5.py [fx] added apex normalization to patched modules (#1300) 2022-07-14 14:24:13 +08:00