mirror of https://github.com/hpcaitech/ColossalAI
0688d92e2d
* fix * padding vocab_size when using pipeline parallellism padding vocab_size when using pipeline parallellism fix fix * fix * fix fix fix * fix gather output * fix * fix * fix fix resize embedding fix resize embedding * fix resize embedding fix * revert * revert * revert * fix lm forward distribution * fix * test ci * fix |
||
---|---|---|
.. | ||
test_adam_kernel.py | ||
test_adam_optim.py | ||
test_lr_scheduler.py | ||
test_nvme.py |