mirror of https://github.com/hpcaitech/ColossalAI
ed4c448488
* fix remaining t5 bugs/rewrite t5 tests * fix multi-tensor communication in pipeline * rearrange test_config * fix keyerror in sync_shared_params * fix get_held_layers & Randomnizer, complete t5 tests * erase printing * fix get_held_layers through modifying _release_unheld_layers * fix _get_recursive_held_layers bug |
||
---|---|---|
.. | ||
__init__.py | ||
_operation.py | ||
dropout.py | ||
embedding.py | ||
linear.py | ||
loss.py | ||
normalization.py | ||
parallel_module.py | ||
qkv_fused_linear.py | ||
utils.py |