mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Missed moving some seresnet -> legacy in sotabench. Check sotabench cache.

Browse Source
This commit is contained in:
Ross Wightman 2020-09-25 15:51:04 -07:00
parent e8e2d9cabf
commit 4be5b51e0a
2 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
sotabench.py
View File

@ -117,11 +117,7 @@ model_list = [
_entry('wide_resnet50_2', 'Wide-ResNet-50', '1605.07146'), _entry('wide_resnet50_2', 'Wide-ResNet-50', '1605.07146'),
_entry('seresnet18', 'SE-ResNet-18', '1709.01507'),
_entry('seresnet34', 'SE-ResNet-34', '1709.01507'),
_entry('seresnet50', 'SE-ResNet-50', '1709.01507'), _entry('seresnet50', 'SE-ResNet-50', '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', '1812.01187', _entry('seresnext26d_32x4d', 'SE-ResNeXt-26-D 32x4d', '1812.01187',
model_desc='Block cfg of SE-ResNeXt-34 w/ Bottleneck, deep stem, and avg-pool in downsample layers.'), 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', '1812.01187', _entry('seresnext26t_32x4d', 'SE-ResNeXt-26-T 32x4d', '1812.01187',
@ -261,9 +257,13 @@ model_list = [
_entry('nasnetalarge', 'NASNet-A Large', '1707.07012', batch_size=BATCH_SIZE // 4), _entry('nasnetalarge', 'NASNet-A Large', '1707.07012', batch_size=BATCH_SIZE // 4),
_entry('pnasnet5large', 'PNASNet-5', '1712.00559', batch_size=BATCH_SIZE // 4), _entry('pnasnet5large', 'PNASNet-5', '1712.00559', batch_size=BATCH_SIZE // 4),
_entry('xception', 'Xception', '1610.02357', batch_size=BATCH_SIZE//2), _entry('xception', 'Xception', '1610.02357', batch_size=BATCH_SIZE//2),
_entry('legacy_seresnet18', 'SE-ResNet-18', '1709.01507'),
_entry('legacy_seresnet34', 'SE-ResNet-34', '1709.01507'),
_entry('legacy_seresnet50', 'SE-ResNet-50', '1709.01507'), _entry('legacy_seresnet50', 'SE-ResNet-50', '1709.01507'),
_entry('legacy_seresnet101', 'SE-ResNet-101', '1709.01507'), _entry('legacy_seresnet101', 'SE-ResNet-101', '1709.01507'),
_entry('legacy_seresnet152', 'SE-ResNet-152', '1709.01507'), _entry('legacy_seresnet152', 'SE-ResNet-152', '1709.01507'),
_entry('legacy_seresnext26_32x4d', 'SE-ResNeXt-26 32x4d', '1709.01507',
model_desc='Block cfg of SE-ResNeXt-34 w/ Bottleneck'),
_entry('legacy_seresnext50_32x4d', 'SE-ResNeXt-50 32x4d', '1709.01507'), _entry('legacy_seresnext50_32x4d', 'SE-ResNeXt-50 32x4d', '1709.01507'),
_entry('legacy_seresnext101_32x4d', 'SE-ResNeXt-101 32x4d', '1709.01507'), _entry('legacy_seresnext101_32x4d', 'SE-ResNeXt-101 32x4d', '1709.01507'),
_entry('legacy_senet154', 'SENet-154', '1709.01507'), _entry('legacy_senet154', 'SENet-154', '1709.01507'),
@ -522,6 +522,9 @@ for m in model_list:
evaluator.add(dict(zip(image_ids, list(output)))) evaluator.add(dict(zip(image_ids, list(output))))
sample_count += num_samples sample_count += num_samples
bar.update(num_samples) bar.update(num_samples)
if evaluator.cache_exists:
break
bar.close() bar.close()
evaluator.save() evaluator.save()

2
sotabench_setup.sh
View File

@ -1,8 +1,10 @@
#!/usr/bin/env bash #!/usr/bin/env bash
source /workspace/venv/bin/activate source /workspace/venv/bin/activate
pip install --upgrade pip
pip install -r requirements-sotabench.txt pip install -r requirements-sotabench.txt
apt-get update
apt-get install -y libjpeg-dev zlib1g-dev libpng-dev libwebp-dev apt-get install -y libjpeg-dev zlib1g-dev libpng-dev libwebp-dev
pip uninstall -y pillow pip uninstall -y pillow
CC="cc -mavx2" pip install -U --force-reinstall pillow-simd CC="cc -mavx2" pip install -U --force-reinstall pillow-simd