Add larger EfficientNet config types.

PiperOrigin-RevId: 304485937

Add larger EfficientNet config types.
PiperOrigin-RevId: 304485937
ce09ca02 · A. Unique TensorFlower · 0dcf4941 · ce09ca02
Commit ce09ca02 authored Apr 02, 2020 by A. Unique TensorFlower
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

official/vision/image_classification/efficientnet/efficientnet_model.py ...n/image_classification/efficientnet/efficientnet_model.py +2 -0

No files found.
--- a/official/vision/image_classification/efficientnet/efficientnet_model.py
+++ b/official/vision/image_classification/efficientnet/efficientnet_model.py
@@ -104,6 +104,8 @@ MODEL_CONFIGS = {
    'efficientnet-b5': ModelConfig.from_args(1.6, 2.2, 456, 0.4),
    'efficientnet-b6': ModelConfig.from_args(1.8, 2.6, 528, 0.5),
    'efficientnet-b7': ModelConfig.from_args(2.0, 3.1, 600, 0.5),
+    'efficientnet-b8': ModelConfig.from_args(2.2, 3.6, 672, 0.5),
+    'efficientnet-l2': ModelConfig.from_args(4.3, 5.3, 800, 0.5),
 }

 CONV_KERNEL_INITIALIZER = {