From 35e835f618b43ef3abef6c3a341bcc27088e939a Mon Sep 17 00:00:00 2001 From: songsenand Date: Fri, 13 Feb 2026 01:31:18 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BD=BF=E7=94=A8=20hint=20=E5=AD=97=E6=AE=B5?= =?UTF-8?q?=E6=9B=BF=E4=BB=A3=E5=8E=9F=E5=A7=8B=20input=5Fids=20=E5=92=8C?= =?UTF-8?q?=20attention=5Fmask=20=E8=BF=9B=E8=A1=8C=E6=8E=A8=E7=90=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/trainer/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/trainer/model.py b/src/trainer/model.py index 723954b..8183dcc 100644 --- a/src/trainer/model.py +++ b/src/trainer/model.py @@ -230,8 +230,8 @@ class MoEModel(nn.Module): with torch.no_grad(): for batch in eval_dataloader: # 移动数据到模型设备 - input_ids = batch["input_ids"].to(self.device) - attention_mask = batch["attention_mask"].to(self.device) + input_ids = batch['hint']["input_ids"].to(self.device) + attention_mask = batch['hint']["attention_mask"].to(self.device) pg = batch["pg"].to(self.device) labels = batch["char_id"].to(self.device)