未验证 提交 052da3d7 编写于 作者: Z Zhong Hui 提交者: GitHub

Quick fix typo (#5273)

上级 82367481
......@@ -327,7 +327,7 @@ class ErnieForTokenClassification(ErniePretrainedModel):
class ErnieForPretraining(ErniePretrainedModel):
def __init__(self, ernie):
super(ErnieForPretraining, self).__init__()
self.ernie = bert
self.ernie = ernie
self.cls = ErniePretrainingHeads(
self.ernie.config["hidden_size"],
self.ernie.config["vocab_size"],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册