mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-01-21 14:13:45 +00:00
6c5996a56e
* add Embedding * Add bert test * polish * add check module test * polish * polish * polish * polish
3 lines
103 B
Python
3 lines
103 B
Python
from .colo_module import ColoModule
|
|
from .linear import ColoLinear
|
|
from .embedding import ColoEmbedding |