We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ce10d11 commit 587125eCopy full SHA for 587125e
1 file changed
FlagEmbedding/abc/finetune/embedder/AbsDataset.py
@@ -130,7 +130,7 @@ def __getitem__(self, item):
130
passages.append(data['neg'][neg_idx])
131
132
if self.args.knowledge_distillation:
133
- assert isinstance(data['pos_scores'], list) and isinstance(data['neg_scores', list])
+ assert isinstance(data['pos_scores'], list) and isinstance(data['neg_scores'], list)
134
teacher_scores.append(data['pos_scores'][pos_idx])
135
for neg_idx in neg_idxs:
136
teacher_scores.append(data['neg_scores'][neg_idx])
0 commit comments