You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_fx/test_tracer/test_hf_model
YuliangLiu0306 942c8cd1fb
[fx] refactor tracer to trace complete graph (#1342)
2 years ago
..
test_hf_albert.py [fx] fixed unit tests for torch 1.12 (#1327) 2 years ago
test_hf_bert.py [fx] fixed unit tests for torch 1.12 (#1327) 2 years ago
test_hf_gpt.py [fx] fixed unit tests for torch 1.12 (#1327) 2 years ago
test_hf_opt.py [fx] refactor tracer to trace complete graph (#1342) 2 years ago
test_hf_t5.py [fx] fixed unit tests for torch 1.12 (#1327) 2 years ago
utils.py