Release EfficientNet model implementation (#475)

Summary: Pull Request resolved: #475 Reviewed By: vreis Differential Revision: D20972639 fbshipit-source-id: 3d68c43774e3384f7799982475296c736ab8a398
facebookresearch · Apr 13, 2020 · d16fc33 · d16fc33
1 parent 3539f57
commit d16fc33
Show file tree

Hide file tree

Showing 3 changed files with 679 additions and 0 deletions.
diff --git a/classy_vision/models/__init__.py b/classy_vision/models/__init__.py
@@ -92,6 +92,7 @@ def build_model(config):
     ClassyModelHeadExecutorWrapper,  # isort:skip
 )  # isort:skip
 from .densenet import DenseNet  # isort:skip
+from .efficientnet import EfficientNet  # isort:skip
 from .mlp import MLP  # isort:skip
 from .resnet import ResNet  # isort:skip
 from .resnext import ResNeXt  # isort:skip
@@ -108,6 +109,7 @@ def build_model(config):
     "ClassyModelHeadExecutorWrapper",
     "ClassyModelWrapper",
     "DenseNet",
+    "EfficientNet",
     "MLP",
     "ResNet",
     "ResNeXt",