-
Notifications
You must be signed in to change notification settings - Fork 87
Add convbert model #1036
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
Add convbert model #1036
Conversation
a66e02b to
69765bf
Compare
bb48d2f to
77f4f17
Compare
| _CONFIG_FOR_DOC = "ConvBertConfig" | ||
|
|
||
|
|
||
| def load_tf_weights_in_convbert(model, config, tf_checkpoint_path): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
同cannie模型,看下是否需要提公共函数
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
位置和原仓是一致的
| import tensorflow as tf | ||
| except ImportError: | ||
| logger.error( | ||
| "Loading a TensorFlow model in PyTorch, requires TensorFlow to be installed. Please see " |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
PyTorch->mindspore
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
|
|
||
|
|
||
| CONVBERT_START_DOCSTRING = r""" | ||
| This model is a PyTorch [nn.Cell](https://www.mindspore.cn/docs/en/r2.5.0/api_python/nn/mindspore.nn.Cell.html) sub-class. Use |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
pytorch->mindspore,全局搜索下
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done
What does this PR do?
Add convbert model
Fixes # (issue)
Adds # (feature)
Before submitting
What's New. Here are thedocumentation guidelines
Who can review?
Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.
@xxx