fixed head format

1 month ago · ef8c59cf50
parent b339394537
commit ef8c59cf50
1 changed files with 1 additions and 1 deletions
--- a/ultralytics/nn/modules/head.py
+++ b/ultralytics/nn/modules/head.py
@ -28,7 +28,7 @@ class Detect(nn.Module):
    shape = None
    anchors = torch.empty(0)  # init
    strides = torch.empty(0)  # init
-    format = "_"
+    format = None
    legacy = False  # backward compatibility for v3/v5/v8/v9 models

    def __init__(self, nc=80, ch=()):