Browse Source

update gelan-s pretrained

yjh0410 1 năm trước cách đây
mục cha
commit
38bfb0e2a1
1 tập tin đã thay đổi với 1 bổ sung3 xóa
  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 ----------------------------