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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-24 22:35:06 +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