Fix Mobilenet V3 model name for sotabench. Minor res2net cleanup.

Author: rwightman

sotabench.py

@@ -1,3 +1,4 @@
+import torch
 from torchbench.image_classification import ImageNet
 from timm import create_model
 from timm.data import resolve_data_config, create_transform
@@ -77,7 +78,7 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
     _entry('mixnet_m', 'MixNet-M', '1907.09595'),
     _entry('mixnet_s', 'MixNet-S', '1907.09595'),
     _entry('mnasnet_100', 'MnasNet-B1', '1807.11626'),
-    _entry('mobilenetv3_100', 'MobileNet V3(1.0)', '1905.02244',
+    _entry('mobilenetv3_100', 'MobileNet V3-Large 1.0', '1905.02244',
            model_desc='Trained in PyTorch with RMSProp, exponential LR decay, and hyper-params matching '
                       'paper as closely as possible.'),
     _entry('resnet18', 'ResNet-18', '1812.01187'),
@@ -216,4 +217,6 @@ def _entry(model_name, paper_model_name, paper_arxiv_id, batch_size=BATCH_SIZE,
         data_root=os.environ.get('IMAGENET_DIR', './imagenet')
     )
 
+    torch.cuda.empty_cache()
+
 

timm/models/res2net.py

@@ -58,7 +58,7 @@ def __init__(self, inplanes, planes, stride=1, downsample=None,
         super(Bottle2neck, self).__init__()
         assert dilation == 1 and previous_dilation == 1  # FIXME support dilation
         self.scale = scale
-        self.is_first = True if stride > 1 or downsample is not None else False
+        self.is_first = stride > 1 or downsample is not None
         self.num_scales = max(1, scale - 1)
         width = int(math.floor(planes * (base_width / 64.0))) * cardinality
         outplanes = planes * self.expansion