Files
ColossalAI/colossalai/tensor/modules/__init__.py
Ziyue Jiang 6c5996a56e [Tensor] add module check and bert test (#1031)
* add Embedding

* Add bert test

* polish

* add check module test

* polish

* polish

* polish

* polish
2022-05-26 18:15:42 +08:00

3 lines
103 B
Python

from .colo_module import ColoModule
from .linear import ColoLinear
from .embedding import ColoEmbedding