rename class in __init__

2019-10-08 17:32:28 +02:00
parent 61ed889005
commit 770b15b58c
1 changed files with 1 additions and 1 deletions
--- a/transformers/modeling_bert.py
+++ b/transformers/modeling_bert.py
@@ -1449,7 +1449,7 @@ class Bert2Rnd(BertPreTrainedModel):
    """

    def __init__(self, config):
-        super(Bert2Bert, self).__init__(config)
+        super(Bert2Rnd, self).__init__(config)
        self.encoder = BertModel(config)
        self.decoder = BertDecoderModel(config)