diff --git a/timm/models/efficientnet.py b/timm/models/efficientnet.py index 2ed2a14e..a4f89055 100644 --- a/timm/models/efficientnet.py +++ b/timm/models/efficientnet.py @@ -1,17 +1,25 @@ -""" Generic EfficientNets +""" PyTorch EfficientNet Family + +An implementation of EfficienNet that covers variety of related models with efficient architectures: + +* EfficientNet (B0-B8 + Tensorflow pretrained AutoAug/RandAug/AdvProp weight ports) + - EfficientNet: Rethinking Model Scaling for CNNs - https://arxiv.org/abs/1905.11946 + - CondConv: Conditionally Parameterized Convolutions for Efficient Inference - https://arxiv.org/abs/1904.04971 + - Adversarial Examples Improve Image Recognition - https://arxiv.org/abs/1911.09665 -A generic class with building blocks to support a variety of models with efficient architectures: -* EfficientNet (B0-B7) -* EfficientNet-EdgeTPU -* EfficientNet-CondConv * MixNet (Small, Medium, and Large) -* MnasNet B1, A1 (SE), Small -* MobileNet V1, V2, and V3 + - MixConv: Mixed Depthwise Convolutional Kernels - https://arxiv.org/abs/1907.09595 + +* MNasNet B1, A1 (SE), Small + - MnasNet: Platform-Aware Neural Architecture Search for Mobile - https://arxiv.org/abs/1807.11626 + * FBNet-C + - FBNet: Hardware-Aware Efficient ConvNet Design via Differentiable NAS - https://arxiv.org/abs/1812.03443 + * Single-Path NAS Pixel1 -* And likely more... + - Single-Path NAS: Designing Hardware-Efficient ConvNets - https://arxiv.org/abs/1904.02877 -TODO not all combinations and variations have been tested. Currently working on training hyper-params... +* And likely more... Hacked together by Ross Wightman """ @@ -183,8 +191,6 @@ default_cfgs = { url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/tf_mixnet_l-6c92e0c8.pth'), } - - _DEBUG = False