Skip to content

Commit

Permalink
should be false, listwise is an edge case
Browse files Browse the repository at this point in the history
  • Loading branch information
Parry-Parry committed Nov 11, 2024
1 parent cb210b2 commit e40c22a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rankers/datasets/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ def __init__(self,
group_size: int = 2,
no_positive: bool = False,
lazy_load_text : bool = True,
top_k_group : bool = True
top_k_group : bool = False,
) -> None:
assert training_dataset_file.endswith('jsonl'), "Training dataset should be a JSONL file and should not be compressed"

Expand Down

0 comments on commit e40c22a

Please sign in to comment.