ColossalAI/tests/test_fx/test_tracer
Frank Lee 085e7f4eff
[test] fixed torchrec registration in model zoo (#3177)
* [test] fixed torchrec registration in model zoo

* polish code

* polish code

* polish code
2023-03-20 16:19:06 +08:00
..
test_hf_model
test_timm_model
test_torchaudio_model
test_torchrec_model [test] fixed torchrec registration in model zoo (#3177) 2023-03-20 16:19:06 +08:00
test_torchvision_model
test_activation_checkpoint_annotation.py
test_bias_addition_module.py
test_control_flow.py
test_functional_conv.py
test_patched_module.py
test_patched_op.py