Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

支持自己搭建的Transformer吗? #248

Open
wapping opened this issue Sep 14, 2021 · 1 comment
Open

支持自己搭建的Transformer吗? #248

wapping opened this issue Sep 14, 2021 · 1 comment

Comments

@wapping
Copy link

wapping commented Sep 14, 2021

是不是仅支持README中提到的几个模型以下几个模型?假设基于pytorch搭建一个transformer模型,能用turbo transformer加速吗?
BERT [Python] [C++]
ALBERT [Python]
Roberta [Python]
Transformer Decoder [Python]
GPT2 [Python]

@feifeibear
Copy link
Collaborator

怎么定义自己搭建transformer?只要是transformers构成的都模型,都可以复用我们写好的transformers接口
https://github.com/Tencent/TurboTransformers/blob/master/turbo_transformers/python/tests/bert_layer_test.py
bert layer就是一个transformer结构

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants