Skip to content

Commit ef30e7f

Browse files
committed
Update sotabench with EffNet Lite models
1 parent 3406e58 commit ef30e7f

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

sotabench.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -191,12 +191,25 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
191191
model_desc='Ported from official Google AI Tensorflow weights'),
192192
_entry('tf_efficientnet_cc_b1_8e', 'EfficientNet-CondConv-B1 8 experts', '1904.04971',
193193
model_desc='Ported from official Google AI Tensorflow weights'),
194+
194195
_entry('tf_efficientnet_es', 'EfficientNet-EdgeTPU-S', '1905.11946',
195196
model_desc='Ported from official Google AI Tensorflow weights'),
196197
_entry('tf_efficientnet_em', 'EfficientNet-EdgeTPU-M', '1905.11946',
197198
model_desc='Ported from official Google AI Tensorflow weights'),
198199
_entry('tf_efficientnet_el', 'EfficientNet-EdgeTPU-L', '1905.11946', batch_size=BATCH_SIZE//2,
199200
model_desc='Ported from official Google AI Tensorflow weights'),
201+
202+
_entry('tf_efficientnet_lite0', 'EfficientNet-Lite0', '1905.11946',
203+
model_desc='Ported from official Google AI Tensorflow weights'),
204+
_entry('tf_efficientnet_lite1', 'EfficientNet-Lite1', '1905.11946',
205+
model_desc='Ported from official Google AI Tensorflow weights'),
206+
_entry('tf_efficientnet_lite2', 'EfficientNet-Lite2', '1905.11946',
207+
model_desc='Ported from official Google AI Tensorflow weights'),
208+
_entry('tf_efficientnet_lite3', 'EfficientNet-Lite3', '1905.11946', batch_size=BATCH_SIZE // 2,
209+
model_desc='Ported from official Google AI Tensorflow weights'),
210+
_entry('tf_efficientnet_lite4', 'EfficientNet-Lite4', '1905.11946', batch_size=BATCH_SIZE // 2,
211+
model_desc='Ported from official Google AI Tensorflow weights'),
212+
200213
_entry('tf_inception_v3', 'Inception V3', '1512.00567', model_desc='Ported from official Tensorflow weights'),
201214
_entry('tf_mixnet_l', 'MixNet-L', '1907.09595', model_desc='Ported from official Google AI Tensorflow weights'),
202215
_entry('tf_mixnet_m', 'MixNet-M', '1907.09595', model_desc='Ported from official Google AI Tensorflow weights'),

0 commit comments

Comments
 (0)