:twisted_rightwards_arrows: [Merge] branch 'MODEL' into TEST

yolo/utils/bounding_box_utils.py

@@ -314,7 +314,8 @@ class Vec2Box:
 
     def create_auto_anchor(self, model: YOLO, image_size):
         W, H = image_size
-        dummy_input = torch.zeros(1, 3, H, W).to(self.device)
+        # TODO: need accelerate dummy test
+        dummy_input = torch.zeros(1, 3, H, W)
         dummy_output = model(dummy_input)
         strides = []
         for predict_head in dummy_output["Main"]: