CogAgent icon indicating copy to clipboard operation
CogAgent copied to clipboard
Published 9 months ago • verified publisher icon zai-org

Update finetune.py

Open work-space-account opened this issue 10 months ago • 1 comments

原有代码逻辑，似乎会在process_batch_eval这一步中将一轮对话中role为user和assistant拆成两轮，后续计算label_token和pred_token的rouge分数的时候，user文本和模型和当前文本预测输出文本做相似度，在compute_metrics里面用奇偶分开读取batched_pred_ids和batched_pred_ids

work-space-account avatar Feb 18 '25 09:02 work-space-account

该问题已经在这个commit中合并 https://github.com/THUDM/CogAgent/pull/39

sixsixcoder avatar Feb 19 '25 10:02 sixsixcoder