Skip to content

Commit ff99625

Browse files
committed
Missed a few models in the model/_all_ list for senet
1 parent 99122aa commit ff99625

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

models/senet.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@
1919
from models.adaptive_avgmax_pool import SelectAdaptivePool2d
2020
from data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
2121

22-
_models = ['senet154', 'seresnet50', 'seresnet101', 'seresnet152',
23-
'seresnext50_32x4d', 'seresnext101_32x4d']
22+
_models = ['seresnet18', 'seresnet34', 'seresnet50', 'seresnet101', 'seresnet152', 'senet154',
23+
'seresnext26_32x4d', 'seresnext50_32x4d', 'seresnext101_32x4d']
2424
__all__ = ['SENet'] + _models
2525

2626

0 commit comments

Comments
 (0)