ColossalAI/tests/test_fx/test_tracer/test_hf_model
Frank Lee 4a09fc0947
[fx] fixed tracing with apex-based T5 model (#1252)
* [fx] fixed tracing with apex-based T5 model

* polish code

* polish code
2022-07-12 15:19:25 +08:00
..
test_hf_albert.py [fx] added testing for all albert variants (#1211) 2022-07-06 15:11:08 +08:00
test_hf_bert.py [fx] added testing for all gpt variants (#1210) 2022-07-06 14:03:13 +08:00
test_hf_gpt.py [fx] added testing for all gpt variants (#1210) 2022-07-06 14:03:13 +08:00
test_hf_opt.py [fx] added testing for all albert variants (#1211) 2022-07-06 15:11:08 +08:00
test_hf_t5.py [fx] fixed tracing with apex-based T5 model (#1252) 2022-07-12 15:19:25 +08:00
utils.py [fx] fixed huggingface OPT and T5 results misalignment (#1227) 2022-07-07 16:29:58 +08:00