423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-15 03:10:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

tiny fix

Browse Source
This commit is contained in:
hiyouga
2024-07-01 05:43:17 +08:00
parent 8c41a0aa6d
commit 73280b7dc7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/train/sft/metric.py
View File

@@ -48,7 +48,7 @@ def compute_accuracy(eval_preds: "EvalPrediction") -> Dict[str, float]:
preds, labels = eval_preds.predictions, eval_preds.label_ids preds, labels = eval_preds.predictions, eval_preds.label_ids
accuracies = [] accuracies = []
for i in range(len(preds)): for i in range(len(preds)):
pred, label = preds[i, 1:], labels[i, :-1] pred, label = preds[i, :-1], labels[i, 1:]
label_mask = label != IGNORE_INDEX label_mask = label != IGNORE_INDEX
accuracies.append(np.mean(pred[label_mask] == label[label_mask])) accuracies.append(np.mean(pred[label_mask] == label[label_mask]))