From ff9962560368f31cf62397ff0c2930b632ef9f48 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 31 May 2019 10:51:18 -0700 Subject: [PATCH] Missed a few models in the model/_all_ list for senet --- models/senet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/senet.py b/models/senet.py index 51844c74..5cfd78f9 100644 --- a/models/senet.py +++ b/models/senet.py @@ -19,8 +19,8 @@ from models.helpers import load_pretrained from models.adaptive_avgmax_pool import SelectAdaptivePool2d from data import IMAGENET_DEFAULT_MEAN, IMAGENET_DEFAULT_STD -_models = ['senet154', 'seresnet50', 'seresnet101', 'seresnet152', - 'seresnext50_32x4d', 'seresnext101_32x4d'] +_models = ['seresnet18', 'seresnet34', 'seresnet50', 'seresnet101', 'seresnet152', 'senet154', + 'seresnext26_32x4d', 'seresnext50_32x4d', 'seresnext101_32x4d'] __all__ = ['SENet'] + _models