BUG SQUASED: removed typo

Spritan · Feb 27, 2024 · b4db934 · b4db934
1 parent a12dd5c
commit b4db934
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/YOLOv8_Explainer/core.py b/YOLOv8_Explainer/core.py
@@ -177,7 +177,7 @@ def __init__(
         Initialize the YOLOv8 heatmap layer.
         """
         device = device
-        self.backward_type = "all"
+        backward_type = "all"
         ckpt = torch.load(weight)
         model_names = ckpt['model'].names
         model = attempt_load_weights(weight, device)