cunliangkong/LlamaFactory
1
0
Fork 0
mirror of https://github.com/hiyouga/LlamaFactory.git synced 2026-03-24 03:03:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

support batch_eval_metrics, fix #4826

Browse Source 
Former-commit-id: 3fe1df17188825f8a32fbe6a1294b4b532ce0c85
This commit is contained in:
hiyouga
2024-07-17 00:33:00 +08:00
parent 45367105fc
commit 8c93921952
7 changed files with 85 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/llamafactory/train/rm/workflow.py
View File

@@ -22,7 +22,7 @@ from ...extras.ploting import plot_loss
from ...model import load_model, load_tokenizer
from ..callbacks import fix_valuehead_checkpoint
from ..trainer_utils import create_modelcard_and_push
from .metric import compute_accuracy
from .metric import ComputeAccuracy
from .trainer import PairwiseTrainer
@@ -55,7 +55,7 @@ def run_rm(
finetuning_args=finetuning_args,
data_collator=data_collator,
callbacks=callbacks,
compute_metrics=compute_accuracy,
compute_metrics=ComputeAccuracy(),
**dataset_module,
**tokenizer_module,
)