Add CSPResNet50 weights, 79.6 top-1 at 256x256

4 years ago · 0f5d9d8166
parent 0734c0d187
commit 0f5d9d8166
1 changed files with 2 additions and 1 deletions
--- a/timm/models/cspnet.py
+++ b/timm/models/cspnet.py
@ -37,7 +37,8 @@ def _cfg(url='', **kwargs):


 default_cfgs = {
-    'cspresnet50': _cfg(url=''),
+    'cspresnet50': _cfg(
+        url='https://github.com/rwightman/pytorch-image-models/releases/download/v0.1-weights/cspresnet50_ra-d3e8d487.pth'),
    'cspresnet50d': _cfg(url=''),
    'cspresnet50w': _cfg(url=''),
    'cspresnext50': _cfg(