From dd28ab45df3b8b89cf445f568426fa944a8ec5d9 Mon Sep 17 00:00:00 2001 From: ElcarimQAQ <1067095308@qq.com> Date: Fri, 18 Apr 2025 14:19:28 +0800 Subject: [PATCH] del excess code --- applications/ColossalChat/coati/distributed/producer.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/applications/ColossalChat/coati/distributed/producer.py b/applications/ColossalChat/coati/distributed/producer.py index 8408693b5..0535fbd78 100644 --- a/applications/ColossalChat/coati/distributed/producer.py +++ b/applications/ColossalChat/coati/distributed/producer.py @@ -240,14 +240,10 @@ class SimpleProducer(BaseProducer): all_accs = torch.tensor([], device=self.device) for test_batch in self.val_dataloader: - # test_batch['input_ids'].size() [32, 300] - # test_batch["gt_answer"] orch.Size([32, 1, 300]) test_output = self.rollout(**test_batch) - # test_output["response_idx"] torch.Size([32, 8, 2]) num_generations = test_output["response_idx"].size(1) print("num_generations", num_generations) data = {k: v.view(-1, v.size(-1)) for k, v in test_output.items()} - # data = test_output reward_group = self.reward_model( data["input_ids"], gt_answer=data["gt_answer"], response_idx=data["response_idx"])