13 lines
370 B
Python
13 lines
370 B
Python
"""
|
|
Source: https://github.com/lukemelas/EfficientNet-PyTorch.
|
|
"""
|
|
__version__ = "0.6.4"
|
|
from .model import (
|
|
EfficientNet, efficientnet_b0, efficientnet_b1, efficientnet_b2,
|
|
efficientnet_b3, efficientnet_b4, efficientnet_b5, efficientnet_b6,
|
|
efficientnet_b7
|
|
)
|
|
from .utils import (
|
|
BlockArgs, BlockDecoder, GlobalParams, efficientnet, get_model_params
|
|
)
|