소스 검색

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