wehub-resource-sync
29 lines
554 B
Python
29 lines
554 B
Python
from transformers import HfArgumentParser
|
|
|
|
from FlagEmbedding.evaluation.msmarco import (
|
|
MSMARCOEvalArgs, MSMARCOEvalModelArgs,
|
|
MSMARCOEvalRunner
|
|
)
|
|
|
|
|
|
def main():
|
|
parser = HfArgumentParser((
|
|
MSMARCOEvalArgs,
|
|
MSMARCOEvalModelArgs
|
|
))
|
|
|
|
eval_args, model_args = parser.parse_args_into_dataclasses()
|
|
eval_args: MSMARCOEvalArgs
|
|
model_args: MSMARCOEvalModelArgs
|
|
|
|
runner = MSMARCOEvalRunner(
|
|
eval_args=eval_args,
|
|
model_args=model_args
|
|
)
|
|
|
|
runner.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|