浏览代码

update gelan-s pretrained

yjh0410 1 年之前
父节点
当前提交
38bfb0e2a1
共有 1 个文件被更改,包括 1 次插入3 次删除
  1. 1 3
      yolo/models/gelan/gelan_backbone.py

+ 1 - 3
yolo/models/gelan/gelan_backbone.py

@@ -9,9 +9,7 @@ except:
 # IN1K pretrained weight
 pretrained_urls = {
     's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/gelan_s_in1k_68.4.pth",
-    'm': None,
-    'l': None,
-    'x': None,
+    'c': None,
 }
 
 # ---------------------------- Basic functions ----------------------------