update model

pull/17/head
KaiyangZhou 2018-05-04 23:14:21 +01:00
parent cd3a565ad0
commit 91c33c7719
2 changed files with 3 additions and 1 deletions

View File

@ -7,7 +7,7 @@ import torch.utils.model_zoo as model_zoo
import os
import sys
__all__ = ['InceptionResNetV2', 'inceptionresnetv2']
__all__ = ['InceptionResNetV2']
pretrained_settings = {
'inceptionresnetv2': {

View File

@ -12,6 +12,7 @@ from .InceptionV4 import *
from .SEResNet import *
from .NASNet import *
from .DPN import *
from .InceptionResNetV2 import *
__factory = {
'resnet50': ResNet50,
@ -27,6 +28,7 @@ __factory = {
'seresnet50': SEResNet50,
'nasnet': NASNetAMobile,
'dpn92': DPN,
'inceptionresnetv2': InceptionResNetV2,
}
def get_names():