Fix: Fix incorrect feature dimension

whats2000 · Aug 2, 2024 · 250cdac · 250cdac
1 parent 3636489
commit 250cdac
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cirr_test_submission_clip.py b/src/cirr_test_submission_clip.py
@@ -249,7 +249,7 @@ def generate_cirr_test_predictions(
 
     # Initialize pairs_id, predicted_features, group_members and reference_names
     pairs_id = []
-    predicted_features = torch.empty((0, 256)).to(device, non_blocking=True)
+    predicted_features = torch.empty((0, clip_text_encoder.text_projection.out_features)).to(device, non_blocking=True)
     group_members = []
     reference_names = []