github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 10:54:49 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4d8a574cd3beb18e497d854e599f2899bc8afa70
ColossalAI/colossalai/tensor/modules/colo_module.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

2.1 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink