From ef30e7f2b6fd1a1a9c4a27e21f4aa09d04ad1f54 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Wed, 18 Mar 2020 13:15:54 -0700 Subject: [PATCH] Update sotabench with EffNet Lite models --- sotabench.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/sotabench.py b/sotabench.py index 7b896819..de3828f4 100644 --- a/sotabench.py +++ b/sotabench.py @@ -191,12 +191,25 @@ model_list = [ model_desc='Ported from official Google AI Tensorflow weights'), _entry('tf_efficientnet_cc_b1_8e', 'EfficientNet-CondConv-B1 8 experts', '1904.04971', model_desc='Ported from official Google AI Tensorflow weights'), + _entry('tf_efficientnet_es', 'EfficientNet-EdgeTPU-S', '1905.11946', model_desc='Ported from official Google AI Tensorflow weights'), _entry('tf_efficientnet_em', 'EfficientNet-EdgeTPU-M', '1905.11946', model_desc='Ported from official Google AI Tensorflow weights'), _entry('tf_efficientnet_el', 'EfficientNet-EdgeTPU-L', '1905.11946', batch_size=BATCH_SIZE//2, model_desc='Ported from official Google AI Tensorflow weights'), + + _entry('tf_efficientnet_lite0', 'EfficientNet-Lite0', '1905.11946', + model_desc='Ported from official Google AI Tensorflow weights'), + _entry('tf_efficientnet_lite1', 'EfficientNet-Lite1', '1905.11946', + model_desc='Ported from official Google AI Tensorflow weights'), + _entry('tf_efficientnet_lite2', 'EfficientNet-Lite2', '1905.11946', + model_desc='Ported from official Google AI Tensorflow weights'), + _entry('tf_efficientnet_lite3', 'EfficientNet-Lite3', '1905.11946', batch_size=BATCH_SIZE // 2, + model_desc='Ported from official Google AI Tensorflow weights'), + _entry('tf_efficientnet_lite4', 'EfficientNet-Lite4', '1905.11946', batch_size=BATCH_SIZE // 2, + model_desc='Ported from official Google AI Tensorflow weights'), + _entry('tf_inception_v3', 'Inception V3', '1512.00567', model_desc='Ported from official Tensorflow weights'), _entry('tf_mixnet_l', 'MixNet-L', '1907.09595', model_desc='Ported from official Google AI Tensorflow weights'), _entry('tf_mixnet_m', 'MixNet-M', '1907.09595', model_desc='Ported from official Google AI Tensorflow weights'),