Can't load custom trained model about yolort

[errx]

diff --git a/yolort/v5/models/yolo.py b/yolort/v5/models/yolo.py
index 38ae7a3..ae75205 100644
--- a/yolort/v5/models/yolo.py
+++ b/yolort/v5/models/yolo.py
@@ -38,7 +38,7 @@ from .experimental import CrossConv, MixConv2d
 if is_module_available("thop"):
     import thop  # for FLOPs computation

-__all__ = ["Model", "Detect"]
+__all__ = ["Model", "Detect", "DetectionModel"]

 LOGGER = logging.getLogger(__name__)

@@ -336,3 +336,5 @@ def parse_model(d, ch):  # model_dict, input_channels(3)
             ch = []
         ch.append(c2)
     return nn.Sequential(*layers), sorted(save)
+
+DetectionModel = Model

this hack seems to be working

[zhiqwang]

Thanks for reporting this bug to us @errx , and seems that your strategy is correct, we do not support YOLOv5's v6.2 or master branch at this time. And it would be great if you could make a Pull Request for this change to us.