소스 검색

调整 efficientnet 默认m

RenLiqiang 7 달 전
부모
커밋
7b80da0bf0
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      models/line_detect/line_net.py

+ 1 - 1
models/line_detect/line_net.py

@@ -338,7 +338,7 @@ _COMMON_META = {
 }
 
 
-def create_efficientnetv2_backbone(name='efficientnet_v2_l', pretrained=True):
+def create_efficientnetv2_backbone(name='efficientnet_v2_m', pretrained=True):
     # 加载EfficientNetV2模型
     if name == 'efficientnet_v2_s':
         weights = EfficientNet_V2_S_Weights.IMAGENET1K_V1 if pretrained else None