add config for yolov5-x

zhiqwang · Dec 29, 2021 · 0934555 · 0934555
1 parent 5dc8996
commit 0934555
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/yolort/v5/helper.py b/yolort/v5/helper.py
@@ -37,6 +37,8 @@ def get_yolov5_size(depth_multiple, width_multiple):
         return "m"
     if depth_multiple == 1.0 and width_multiple == 1.0:
         return "l"
+    if depth_multiple == 1.33 and width_multiple == 1.25:
+        return "x"
     raise NotImplementedError(
         f"Currently does't support architecture with depth: {depth_multiple} "
         f"and {width_multiple}, fell free to create a ticket labeled enhancement to us"