search

OpenBMB / ModelCenter

Efficient, Low-Resource, Distributed transformer implementation based on BMTrain
https://modelcenter.readthedocs.io
Apache License 2.0
233 stars 28 forks source link

vit for bmtrain #11

Closed qyc-98 closed 2 years ago

qyc-98 commented 2 years ago

https://github.com/cqylv/ModelCenter/blob/main/model_center/model/vit_bmtrain.py