ColossalAI/colossalai/nn/layer/parallel_1d at 71e72c48907195096ef02be73e1c5b0feea2653d - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00

Files

History

ver217 10dd8226b1 add gather_output for VocabParallelClassifier1D (#1569 )

2022-09-08 16:40:56 +08:00

..

__init__.py

[model checkpoint] updated saving/loading for 1d layers (#594 )

2022-04-01 16:51:52 +08:00

_operation.py

Refactored docstring to google style

2022-03-29 17:17:47 +08:00

_utils.py

[Tensor] 1d row embedding (#1075 )

2022-06-08 12:04:59 +08:00

layers.py

add gather_output for VocabParallelClassifier1D (#1569 )

2022-09-08 16:40:56 +08:00