update model
parent
cd3a565ad0
commit
91c33c7719
|
@ -7,7 +7,7 @@ import torch.utils.model_zoo as model_zoo
|
|||
import os
|
||||
import sys
|
||||
|
||||
__all__ = ['InceptionResNetV2', 'inceptionresnetv2']
|
||||
__all__ = ['InceptionResNetV2']
|
||||
|
||||
pretrained_settings = {
|
||||
'inceptionresnetv2': {
|
||||
|
|
|
@ -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():
|
||||
|
|
Loading…
Reference in New Issue