mirror of https://github.com/hpcaitech/ColossalAI
eb7742a4bb
* [Gemini] more tests for Gemini * polish code |
||
---|---|---|
.. | ||
utils | ||
__init__.py | ||
bert.py | ||
gpt.py | ||
inline_op_model.py | ||
nested_model.py | ||
no_leaf_module.py | ||
registry.py | ||
repeated_computed_layer.py | ||
resnet.py | ||
simple_net.py |