From 5d3a505d03dc96020ee564050f1d5c4396823980 Mon Sep 17 00:00:00 2001
From: rrjia <360442573@qq.com>
Date: Tue, 19 Jul 2022 17:09:16 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=A4=E4=B8=AA=E5=BC=A0=E9=87=8F=E9=83=BD?=
 =?UTF-8?q?=E8=A6=81=E5=90=8C=E6=AD=A5=E5=88=B0cpu=E4=B8=8A=E6=9D=A5?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 projects/FastRetri/fastretri/retri_evaluator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/FastRetri/fastretri/retri_evaluator.py b/projects/FastRetri/fastretri/retri_evaluator.py
index ccf1a55..b1431d5 100644
--- a/projects/FastRetri/fastretri/retri_evaluator.py
+++ b/projects/FastRetri/fastretri/retri_evaluator.py
@@ -97,7 +97,7 @@ class RetriEvaluator(DatasetEvaluator):
 
     def process(self, inputs, outputs):
         self.features.append(outputs.cpu())
-        self.labels.extend(inputs["targets"])
+        self.labels.extend(inputs["targets"].cpu())
 
     def evaluate(self):
         if comm.get_world_size() > 1: