Skip to content

GitLab

Commit ce09ca02 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Add larger EfficientNet config types. 

PiperOrigin-RevId: 304485937
parent 0dcf4941
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/vision/image_classification/efficientnet/efficientnet_model.py
View file @ ce09ca02
...@@ -104,6 +104,8 @@ MODEL_CONFIGS = { ...@@ -104,6 +104,8 @@ MODEL_CONFIGS = {
'efficientnet-b5': ModelConfig.from_args(1.6, 2.2, 456, 0.4), '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-b6': ModelConfig.from_args(1.8, 2.6, 528, 0.5),
'efficientnet-b7': ModelConfig.from_args(2.0, 3.1, 600, 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 = { CONV_KERNEL_INITIALIZER = {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment