🔧 fix small bug

kaylode · May 4, 2023 · 86c4af3 · 86c4af3
1 parent d994ce0
commit 86c4af3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/theseus/cv/classification/metrics/projection.py b/theseus/cv/classification/metrics/projection.py
@@ -44,7 +44,8 @@ def update(self, outputs: Dict[str, Any], batch: Dict[str, Any]):
             torch.argmax(outputs["outputs"].detach().cpu(), dim=1).numpy().tolist()
         )
         inputs = batch["inputs"]
-        targets = batch["targets"].numpy().tolist()
+        if self.has_labels:
+            targets = batch["targets"].numpy().tolist()
         img_names = batch["img_names"]
 
         for i, _ in enumerate(features):