fix a bug

jpthu17 · Nov 16, 2023 · 63ad5cd · 63ad5cd
1 parent d5716c3
commit 63ad5cd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/HBI/models/banzhaf.py b/HBI/models/banzhaf.py
@@ -126,6 +126,7 @@ def banzhaf_interaction(self, retrieve_logits, text_mask, video_mask, text_weigh
 
         #########################
         _text_mask[:, i] = 1
+        _video_mask[:, j] = 0
 
         _text_weight0, _video_weight0 = text_weight.clone(), video_weight.clone()
         _retrieve_logits0 = retrieve_logits.clone()
@@ -146,6 +147,7 @@ def banzhaf_interaction(self, retrieve_logits, text_mask, video_mask, text_weigh
         banzhaf_value2 = (t2v_logits + v2t_logits) / 2.0
 
         #########################
+        _text_mask[:, i] = 0
         _video_mask[:, j] = 1
 
         _text_weight0, _video_weight0 = text_weight.clone(), video_weight.clone()