Commit 770b15b5 authored by Rémi Louf's avatar Rémi Louf
Browse files

rename class in __init__

parent 61ed8890
...@@ -1449,7 +1449,7 @@ class Bert2Rnd(BertPreTrainedModel): ...@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment