Skip to content

Commit 8ceceef

Browse files
committed
Densenet should default to bicubic interpolation, update model links to 'cpu fix' from long ago
1 parent a4516fe commit 8ceceef

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

models/densenet.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,17 +18,17 @@
1818
def _cfg(url=''):
1919
return {
2020
'url': url, 'num_classes': 1000, 'input_size': (3, 224, 244), 'pool_size': (7, 7),
21-
'crop_pct': 0.875, 'interpolation': 'bilinear',
21+
'crop_pct': 0.875, 'interpolation': 'bicubic',
2222
'mean': IMAGENET_DEFAULT_MEAN, 'std': IMAGENET_DEFAULT_STD,
2323
'first_conv': 'features.conv0', 'classifier': 'classifier',
2424
}
2525

2626

2727
default_cfgs = {
28-
'densenet121': _cfg(url='https://download.pytorch.org/models/densenet121-241335ed.pth'),
29-
'densenet169': _cfg(url='https://download.pytorch.org/models/densenet169-6f0f7f60.pth'),
30-
'densenet201': _cfg(url='https://download.pytorch.org/models/densenet201-4c113574.pth'),
31-
'densenet161': _cfg(url='https://download.pytorch.org/models/densenet161-17b70270.pth'),
28+
'densenet121': _cfg(url='https://download.pytorch.org/models/densenet121-a639ec97.pth'),
29+
'densenet169': _cfg(url='https://download.pytorch.org/models/densenet169-b2777c0a.pth'),
30+
'densenet201': _cfg(url='https://download.pytorch.org/models/densenet201-c1103571.pth'),
31+
'densenet161': _cfg(url='https://download.pytorch.org/models/densenet161-8d451a50.pth'),
3232
}
3333

3434

0 commit comments

Comments
 (0)