RashiAgarwal commited on
Commit
3b48481
1 Parent(s): ba0213c

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -661,7 +661,7 @@ class YoloCAM(BaseCAM):
661
  if self.compute_input_gradient:
662
  input_tensor = torch.autograd.Variable(input_tensor, requires_grad=True)
663
 
664
- outputs = self.activations_and_grads(input_tensor)
665
  if targets is None:
666
  bboxes = [[] for _ in range(1)]
667
  for i in range(3):
 
661
  if self.compute_input_gradient:
662
  input_tensor = torch.autograd.Variable(input_tensor, requires_grad=True)
663
 
664
+ outputs = self.activations_and_gradients(input_tensor)
665
  if targets is None:
666
  bboxes = [[] for _ in range(1)]
667
  for i in range(3):