From 33ab98866922e9ffa8f2d7b4a0bc727d2d66c128 Mon Sep 17 00:00:00 2001 From: Tingquan Gao <35441050@qq.com> Date: Tue, 5 Jan 2021 17:20:40 +0800 Subject: [PATCH] Add EfficientNetB0_small (#539) --- ppcls/modeling/architectures/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/modeling/architectures/__init__.py b/ppcls/modeling/architectures/__init__.py index 741bde96..c7e5aaef 100644 --- a/ppcls/modeling/architectures/__init__.py +++ b/ppcls/modeling/architectures/__init__.py @@ -25,7 +25,7 @@ from .se_resnext import SE_ResNeXt50_32x4d, SE_ResNeXt101_32x4d, SE_ResNeXt152_6 from .dpn import DPN68, DPN92, DPN98, DPN107, DPN131 from .densenet import DenseNet121, DenseNet161, DenseNet169, DenseNet201, DenseNet264 from .hrnet import HRNet_W18_C, HRNet_W30_C, HRNet_W32_C, HRNet_W40_C, HRNet_W44_C, HRNet_W48_C, HRNet_W60_C, HRNet_W64_C, SE_HRNet_W18_C, SE_HRNet_W30_C, SE_HRNet_W32_C, SE_HRNet_W40_C, SE_HRNet_W44_C, SE_HRNet_W48_C, SE_HRNet_W60_C, SE_HRNet_W64_C -from .efficientnet import EfficientNetB0, EfficientNetB1, EfficientNetB2, EfficientNetB3, EfficientNetB4, EfficientNetB5, EfficientNetB6, EfficientNetB7 +from .efficientnet import EfficientNetB0, EfficientNetB1, EfficientNetB2, EfficientNetB3, EfficientNetB4, EfficientNetB5, EfficientNetB6, EfficientNetB7, EfficientNetB0_small from .resnest import ResNeSt50_fast_1s1x64d, ResNeSt50, ResNeSt101 from .googlenet import GoogLeNet from .ghostnet import GhostNet_x0_5, GhostNet_x1_0, GhostNet_x1_3 -- GitLab