Add updated model checkpoint from dingyiwei

zhiqwang · Mar 3, 2021 · fa192f2 · fa192f2
1 parent c29e43f
commit fa192f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/yolort/models/yolo.py b/yolort/models/yolo.py
@@ -135,7 +135,7 @@ def forward(
     'yolov5_darknet_pan_s_r40_coco': f'{model_urls_root}/yolov5_darknet_pan_s_r40_coco-e3fd213d.pt',
     'yolov5_darknet_pan_m_r40_coco': f'{model_urls_root}/yolov5_darknet_pan_m_r40_coco-d295cb02.pt',
     'yolov5_darknet_pan_l_r40_coco': f'{model_urls_root}/yolov5_darknet_pan_l_r40_coco-4416841f.pt',
-    'yolov5_darknet_pan_s_tr_coco': None,
+    'yolov5_darknet_pan_s_tr_coco': f'{model_urls_root}/yolov5_darknet_pan_s_tr_coco-f09f21f7.pt',
 }