SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

rename class in __init__

Browse Source
This commit is contained in:
Rémi Louf
2019-10-08 17:32:28 +02:00
parent 61ed889005
commit 770b15b58c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
transformers/modeling_bert.py
View File

@@ -1449,7 +1449,7 @@ class Bert2Rnd(BertPreTrainedModel):
""" """
def __init__(self, config): def __init__(self, config):
super(Bert2Bert, self).__init__(config) super(Bert2Rnd, self).__init__(config)
self.encoder = BertModel(config) self.encoder = BertModel(config)
self.decoder = BertDecoderModel(config) self.decoder = BertDecoderModel(config)