Skip to content

Commit 1efab2f

Browse files
committed
Fix xception state dict
Parameters of size (d,h,w) are now (d,1,h,w). It did work for previous pytorch versions (with a warning), but not anymore.
1 parent 4667eba commit 1efab2f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pretrainedmodels/models/xception.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
pretrained_settings = {
3434
'xception': {
3535
'imagenet': {
36-
'url': 'http://data.lip6.fr/cadene/pretrainedmodels/xception-b0b7af25.pth',
36+
'url': 'http://data.lip6.fr/cadene/pretrainedmodels/xception-b5690688.pth',
3737
'input_space': 'RGB',
3838
'input_size': [3, 299, 299],
3939
'input_range': [0, 1],
@@ -230,4 +230,4 @@ def xception(num_classes=1000, pretrained='imagenet'):
230230
# TODO: ugly
231231
model.last_linear = model.fc
232232
del model.fc
233-
return model
233+
return model

0 commit comments

Comments
 (0)