Missed a few models in the model/_all_ list for senet

pull/6/head
Ross Wightman 6 years ago
parent 99122aac1c
commit ff99625603

@ -19,8 +19,8 @@ from models.helpers import load_pretrained
from models.adaptive_avgmax_pool import SelectAdaptivePool2d from models.adaptive_avgmax_pool import SelectAdaptivePool2d
from data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD from data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD
_models = ['senet154', 'seresnet50', 'seresnet101', 'seresnet152', _models = ['seresnet18', 'seresnet34', 'seresnet50', 'seresnet101', 'seresnet152', 'senet154',
'seresnext50_32x4d', 'seresnext101_32x4d'] 'seresnext26_32x4d', 'seresnext50_32x4d', 'seresnext101_32x4d']
__all__ = ['SENet'] + _models __all__ = ['SENet'] + _models

Loading…
Cancel
Save