github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/nn/loss
History
アマデウス 9ee197d0e9 moved env variables to global variables; (#215) 
added branch context;
added vocab parallel layers;
moved split_batch from load_batch to tensor parallel embedding layers;
updated gpt model;
updated unit test cases;
fixed few collective communicator bugs
 		2022-02-15 11:31:13 +08:00
..
__init__.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
loss_1d.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
loss_2d.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
loss_2p5d.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
loss_3d.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
loss_moe.py Fixed docstring in colossalai (#171) 2022-01-21 10:44:30 +08:00