两个张量都要同步到cpu上来

2022-07-19 17:09:16 +08:00 · 2022-07-19 17:09:16 +08:00 · 5d3a505d03
parent 4508251d74
commit 5d3a505d03
1 changed files with 1 additions and 1 deletions
--- 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: