fix md argument removal

conservationtechlab · Jul 24, 2024 · 49f8f9f · 49f8f9f
1 parent 7f786e1
commit 49f8f9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/animl/models/yolo.py b/src/animl/models/yolo.py
@@ -147,7 +147,7 @@ def _forward_augment(self, x):
         y = self._clip_augmented(y)  # clip augmented tails
         return torch.cat(y, 1), None  # augmented inference, train
 
-    def _forward_once(self, x, profile=False):
+    def _forward_once(self, x, profile=False, visualize=False):
         y, dt = [], []  # outputs
         for m in self.model:
             if m.f != -1:  # if not from previous layer