浏览代码

train YOLOv7-AF-T

yjh0410 1 年之前
父节点
当前提交
8bb6cd1da8
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      yolo/models/yolov7_af/yolov7_af_backbone.py

+ 1 - 1
yolo/models/yolov7_af/yolov7_af_backbone.py

@@ -8,7 +8,7 @@ except:
 
 
 # IN1K pretrained weight
 # IN1K pretrained weight
 pretrained_urls = {
 pretrained_urls = {
-    't': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elannet_t_in1k_6.6.pth",
+    't': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elannet_t_in1k_62.0.pth",
     'l': None,
     'l': None,
     'x': None,
     'x': None,
 }
 }