We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f03a5b8 commit d22f0aeCopy full SHA for d22f0ae
faq_bert_finetuning.py
@@ -58,7 +58,7 @@ def __init__(self, loss_type="triplet", query_type='faq', neg_type='simple', ver
58
59
self.bert_model = None
60
if self.loss_type == "triplet":
61
- word_embedding_model = models.BERT(self.pre_trained_name)
+ word_embedding_model = models.Transformer(self.pre_trained_name)
62
63
# # Apply mean pooling to get one fixed sized sentence vector
64
pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(),
0 commit comments