mirror of https://github.com/hpcaitech/ColossalAI
![]() * add pipeline policy and bert forward to be done * add bertmodel pipeline forward and make tests * add Bert_Policy and test for policy * update formatting * update formatting * update the code * fix bugs * fix name confilt * add bloom model and policy ,revise the base class of policy * revise * revision * add bert_for_pretraining |
||
---|---|---|
.. | ||
__init__.py | ||
base.py | ||
bert.py | ||
bloom.py |