Skip to content

GitLab

Commit 825375cf authored by Mostofa Patwary's avatar Mostofa Patwary
Browse files

updated the evaluation script for retriver

parent a41e4781
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
tasks/main.py
View file @ 825375cf
...@@ -77,18 +77,12 @@ def get_tasks_args(parser): ...@@ -77,18 +77,12 @@ def get_tasks_args(parser):
# parameters for Av.rank validation method # parameters for Av.rank validation method
# Following options/arguments have been taken directly from DPR codebase # Following options/arguments have been taken directly from DPR codebase
#group.add_argument("--val-av-rank-start-epoch", type=int, default=10000,
# help="Av.rank validation: the epoch from which to enable this validation")
group.add_argument('--val-av-rank-hard-neg', type=int, default=30, group.add_argument('--val-av-rank-hard-neg', type=int, default=30,
help='Av.rank validation: how many hard negatives to' help='Av.rank validation: how many hard negatives to'
' take from each question pool') ' take from each question pool')
group.add_argument('--val-av-rank-other-neg', type=int, default=30, group.add_argument('--val-av-rank-other-neg', type=int, default=30,
help='Av.rank validation: how many other negatives to' help='Av.rank validation: how many other negatives to'
' take from each question pool') ' take from each question pool')
#group.add_argument("--val-av-rank-bsz", type=int, default=128,
# help="Av.rank validation: batch size to process passages")
#group.add_argument("--val-av-rank-max-qs", type=int, default=10000,
# help="Av.rank validation: max num of questions")
return parser return parser
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment