diff --git a/theseus/cv/classification/metrics/projection.py b/theseus/cv/classification/metrics/projection.py index beaec98..2f7dba0 100644 --- 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):