Skip to content
This repository has been archived by the owner on Jul 1, 2024. It is now read-only.

Release EfficientNet model implementation #475

Closed
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions classy_vision/models/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -108,6 +109,7 @@ def build_model(config):
"ClassyModelHeadExecutorWrapper",
"ClassyModelWrapper",
"DenseNet",
"EfficientNet",
"MLP",
"ResNet",
"ResNeXt",
Expand Down
Loading