Spaces:

uw-insight-lab
/

Probing-Vis-Literacy-of-VLMs

Paused

AustingDong commited on Mar 10

Commit

22993a5

1 Parent(s): 492bfa4

Update cam.py

Files changed (1) hide show

demo/cam.py CHANGED Viewed

@@ -344,7 +344,7 @@ class AttentionGuidedCAMLLaVA(AttentionGuidedCAM):
         # Forward pass
         torch.autograd.set_detect_anomaly(True)
-        for param in self.model.named_parameters():
             param.requires_grad = False
         for layer in self.target_layers:
@@ -477,7 +477,7 @@ class AttentionGuidedCAMChartGemma(AttentionGuidedCAM):
         # Forward pass
         torch.autograd.set_detect_anomaly(True)
-        for param in self.model.named_parameters():
             param.requires_grad = False
         for layer in self.target_layers:

         # Forward pass
         torch.autograd.set_detect_anomaly(True)
+        for param in self.model.parameters():
             param.requires_grad = False
         for layer in self.target_layers:
         # Forward pass
         torch.autograd.set_detect_anomaly(True)
+        for param in self.model.parameters():
             param.requires_grad = False
         for layer in self.target_layers: