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_pipeline/test_timm_model
YuliangLiu0306 942c8cd1fb
[fx] refactor tracer to trace complete graph (#1342)
2 years ago
..
test_timm.py [fx] refactor tracer to trace complete graph (#1342) 2 years ago
timm_utils.py [fx]add split module pass and unit test from pipeline passes (#1242) 2 years ago