ColossalAI/tests/components_to_test
Ziyue Jiang dfaff4e243
[Tensor] fix test_model (#916)
* polish test_model

* polish
2022-05-06 18:06:22 +08:00
..
utils
__init__.py
bert.py [Tensor] fix test_model (#916) 2022-05-06 18:06:22 +08:00
nested_model.py
no_leaf_module.py
registry.py
repeated_computed_layer.py
resnet.py
simple_net.py