Spaces:

Sijuade
/

YOLOV3-GradCAM

Runtime error

Sijuade commited on Aug 25, 2023

Commit

94564e5

1 Parent(s): 7cb2d2d

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -25,8 +25,8 @@ config = [
     ["B", 8],
     (1024, 3, 2),
     ["B", 4],  # To this point is Darknet-53
-    (1024, 1, 1),
-    (2048, 3, 1),
     "S",
     (256, 1, 1),
     "U",

     ["B", 8],
     (1024, 3, 2),
     ["B", 4],  # To this point is Darknet-53
+    (512, 1, 1),
+    (1024, 3, 1),
     "S",
     (256, 1, 1),
     "U",