diff --git a/sotabench.py b/sotabench.py index 5f6345f5..3174f330 100644 --- a/sotabench.py +++ b/sotabench.py @@ -46,7 +46,14 @@ model_list = [ ttp=True, args=dict(img_size=320), batch_size=BATCH_SIZE//4), _entry('efficientnet_b0', 'EfficientNet-B0', '1905.11946'), _entry('efficientnet_b1', 'EfficientNet-B1', '1905.11946'), - _entry('efficientnet_b2', 'EfficientNet-B2', '1905.11946'), + _entry('efficientnet_b2', 'EfficientNet-B2', '1905.11946', + model_desc='Trained from scratch in PyTorch w/ RandAugment'), + _entry('efficientnet_b2a', 'EfficientNet-B2 (288x288, 1.0 crop)', '1905.11946', + model_desc='Trained from scratch in PyTorch w/ RandAugment'), + _entry('efficientnet_b3', 'EfficientNet-B3', '1905.11946', + model_desc='Trained from scratch in PyTorch w/ RandAugment'), + _entry('efficientnet_b3a', 'EfficientNet-B3 (320x320, 1.0 crop)', '1905.11946', + model_desc='Trained from scratch in PyTorch w/ RandAugment'), _entry('fbnetc_100', 'FBNet-C', '1812.03443', model_desc='Trained in PyTorch with RMSProp, exponential LR decay'), _entry('gluon_inception_v3', 'Inception V3', '1512.00567', model_desc='Ported from GluonCV Model Zoo'), @@ -95,7 +102,11 @@ model_list = [ _entry('seresnet18', 'SE-ResNet-18', '1709.01507'), _entry('seresnet34', 'SE-ResNet-34', '1709.01507'), _entry('seresnext26_32x4d', 'SE-ResNeXt-26 32x4d', '1709.01507', + model_desc='Block cfg of SE-ResNeXt-34 w/ Bottleneck'), + _entry('seresnext26d_32x4d', 'SE-ResNeXt-26-D 32x4d', '', model_desc='Block cfg of SE-ResNeXt-34 w/ Bottleneck, deep stem, and avg-pool in downsample layers.'), + _entry('seresnext26t_32x4d', 'SE-ResNeXt-26-T 32x4d', '', + model_desc='Block cfg of SE-ResNeXt-34 w/ Bottleneck, deep tiered stem, and avg-pool in downsample layers.'), _entry('spnasnet_100', 'Single-Path NAS', '1904.02877', model_desc='Trained in PyTorch with SGD, cosine LR decay'), _entry('tf_efficientnet_b0', 'EfficientNet-B0 (AutoAugment)', '1905.11946',