Spaces:

uw-insight-lab
/

Probing-Vis-Literacy-of-VLMs

Paused

AustingDong commited on Mar 6

Commit

cd7b50a

1 Parent(s): 09b0453

Update cam.py

Files changed (1) hide show

demo/cam.py CHANGED Viewed

@@ -344,8 +344,7 @@ class AttentionGuidedCAMLLaVA(AttentionGuidedCAM):
         outputs_raw = self.model(**inputs)
         self.model.zero_grad()
-        print(outputs_raw)
-        # loss = self.target_layers[-1].attention_map.sum()
         loss = outputs_raw.logits.max(dim=-1).values.sum()
         loss.backward()

         outputs_raw = self.model(**inputs)
         self.model.zero_grad()
+        print("output_raw", outputs_raw)
         loss = outputs_raw.logits.max(dim=-1).values.sum()
         loss.backward()