Spaces:

RashiAgarwal
/

YOLO_v3_PyTorchLightning

Paused

RashiAgarwal commited on Aug 20, 2023

Commit

acf7440

1 Parent(s): 7b271ac

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -652,7 +652,7 @@ class YoloCAM(BaseCAM):
         self,
         input_tensor: torch.Tensor,
         scaled_anchors: torch.Tensor,
-        targets: List[torch.nn.Module],
         eigen_smooth: bool = False,
     ) -> np.ndarray:
         if self.cuda:

         self,
         input_tensor: torch.Tensor,
         scaled_anchors: torch.Tensor,
+        targets: list[torch.nn.Module],
         eigen_smooth: bool = False,
     ) -> np.ndarray:
         if self.cuda: