Update modeling_provence.py
Browse files- modeling_provence.py +0 -2
modeling_provence.py
CHANGED
|
@@ -226,8 +226,6 @@ class Provence(DebertaV2PreTrainedModel):
|
|
| 226 |
else:
|
| 227 |
selected_contexts[i][j] = selected_contexts[i][j][0]
|
| 228 |
if reorder:
|
| 229 |
-
print(reranking_scores[qi])
|
| 230 |
-
print(np.sort(reranking_scores[i])[::-1][:top_k])
|
| 231 |
idxs = np.argsort(reranking_scores[i])[::-1][:top_k]
|
| 232 |
selected_contexts[i] = [selected_contexts[i][j] for j in idxs]
|
| 233 |
reranking_scores[i] = [reranking_scores[i][j] for j in idxs]
|
|
|
|
| 226 |
else:
|
| 227 |
selected_contexts[i][j] = selected_contexts[i][j][0]
|
| 228 |
if reorder:
|
|
|
|
|
|
|
| 229 |
idxs = np.argsort(reranking_scores[i])[::-1][:top_k]
|
| 230 |
selected_contexts[i] = [selected_contexts[i][j] for j in idxs]
|
| 231 |
reranking_scores[i] = [reranking_scores[i][j] for j in idxs]
|