mirror of https://github.com/hpcaitech/ColossalAI
f3bcc292c8
* move bert related pipeline components to shardformer * fix bugs * revision * fix bert model tests * fix bert_lm_head model tests * fix tests * fix tests * done checks * skip bloom |
||
---|---|---|
.. | ||
test_layer | ||
test_model | ||
__init__.py | ||
test_shard_utils.py | ||
test_with_torch_ddp.py |