Переглянути джерело

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 ----------------------------