Procházet zdrojové kódy

update gelan-s pretrained

yjh0410 před 1 rokem
rodič
revize
38bfb0e2a1
1 změnil soubory, kde provedl 1 přidání a 3 odebrání
  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 ----------------------------