Skip to content

Commit d22f0ae

Browse files
committed
Update loading BERT pretrained model
1 parent f03a5b8 commit d22f0ae

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

faq_bert_finetuning.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def __init__(self, loss_type="triplet", query_type='faq', neg_type='simple', ver
5858

5959
self.bert_model = None
6060
if self.loss_type == "triplet":
61-
word_embedding_model = models.BERT(self.pre_trained_name)
61+
word_embedding_model = models.Transformer(self.pre_trained_name)
6262

6363
# # Apply mean pooling to get one fixed sized sentence vector
6464
pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(),

0 commit comments

Comments
 (0)