update

AyushExel · Aug 30, 2022 · d2af8e1 · d2af8e1
1 parent c7a2ec9
commit d2af8e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/tf.py b/models/tf.py
@@ -330,7 +330,7 @@ def __init__(self, nc=80, anchors=(), nm=32, npr=256, ch=(), imgsz=(640, 640), w
         self.proto = TFProto(ch[0], self.npr, self.nm, w=w.proto)  # protos
         self.detect = TFDetect.call
 
-    def forward(self, x):
+    def call(self, x):
         p = self.proto(x[0])
         x = self.detect(self, x)
         return (x, p) if self.training else (x[0], p) if self.export else (x[0], (x[1], p))