Spaces:

uw-insight-lab
/

Probing-Vis-Literacy-of-VLMs

Paused

AustingDong commited on Mar 6

Commit

8f98bd6

1 Parent(s): 360591a

Update cam.py

Files changed (1) hide show

demo/cam.py CHANGED Viewed

@@ -344,7 +344,8 @@ class AttentionGuidedCAMLLaVA(AttentionGuidedCAM):
         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()

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