25 lines
818 B
Python
25 lines
818 B
Python
from dataclasses import dataclass, field
|
|
from FlagEmbedding.abc.finetune.embedder import AbsEmbedderDataArguments
|
|
|
|
from FlagEmbedding.abc.finetune.embedder import (
|
|
AbsEmbedderTrainingArguments,
|
|
AbsEmbedderModelArguments
|
|
)
|
|
|
|
|
|
@dataclass
|
|
class IREmbedderTrainingArguments(AbsEmbedderTrainingArguments):
|
|
"""
|
|
Training argument class for M3.
|
|
"""
|
|
training_type: str = field(default='retrieval_answer', metadata={"help": "whether to use answer"})
|
|
answer_temperature: float = field(default=None, metadata={"help": "temperature for answer"})
|
|
normalize_answer: bool = field(default=True, metadata={"help": "normalize answer"})
|
|
|
|
@dataclass
|
|
class IREmbedderDataArguments(AbsEmbedderDataArguments):
|
|
"""
|
|
Data argument class for M3.
|
|
"""
|
|
answer_inbatch: bool = field(default=False)
|