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 |
||
---|---|---|
.. | ||
mixed_precision | ||
plugin | ||
__init__.py | ||
accelerator.py | ||
booster.py |