ColossalAI/tests/test_fx
Frank Lee 11973d892d
[fx] added torchvision model tracing testing (#1216)
* [fx] added torchvision model tracing testing

* remove unused imports
2022-07-06 21:37:56 +08:00
..
test_tracer [fx] added torchvision model tracing testing (#1216) 2022-07-06 21:37:56 +08:00
test_coloproxy.py [refactor] move process group from _DistSpec to ColoTensor. (#1203) 2022-07-06 16:15:16 +08:00
test_parallel_1d.py [fx]add autoparallel passes (#1121) 2022-06-15 16:36:46 +08:00
test_pipeline_passes.py [fx]add uniform policy (#1208) 2022-07-06 13:48:11 +08:00