From 770b15b58ceb66a5da72d8030e9aff05fd50848a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?R=C3=A9mi=20Louf?= Date: Tue, 8 Oct 2019 17:32:28 +0200 Subject: [PATCH] rename class in __init__ --- transformers/modeling_bert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformers/modeling_bert.py b/transformers/modeling_bert.py index 258e4c3430..fc698c772e 100644 --- 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)