Merge pull request #61 from tanganke/tanganke-patch-1

Update clip_dataset.py
tanganke · Jan 5, 2025 · 5c73d73 · 5c73d73
2 parents 7316283 + 350c2dc
commit 5c73d73
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/fusion_bench/dataset/clip_dataset.py b/fusion_bench/dataset/clip_dataset.py
@@ -65,4 +65,7 @@ def __getitem__(self, idx: int):
         else:
             # if processor is None, return the raw image directly
             inputs = image
+        # convert boolean label to int, this is for the case when the label is a binary classification task
+        if isinstance(item["label"], bool):
+            item["label"] = 1 if item["label"] else 0
         return inputs, item["label"]