Update __init__.py

zhiqwang · zhiqwang · Jun 14, 2021 · May 25, 2021 · May 25, 2021 · May 25, 2021
commit 50ff49d4d113996850e2d2912618ed66dc0bf1b9
diff --git a/yolort/models/__init__.py b/yolort/models/__init__.py
@@ -9,7 +9,7 @@
 from typing import Any
 
 
-def yolov5s(upstream_version: str = 'r4.0', export_friendly: bool = False,use_square_box = False, **kwargs: Any):
+def yolov5s(upstream_version: str = 'r4.0', export_friendly: bool = False,use_square_box: bool = False, **kwargs: Any):
     """
     Args:
         upstream_version (str): model released by the upstream YOLOv5. Possible values