diff --git a/models/__init__.py b/models/__init__.py index 7f67a9e55..877893903 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -1,14 +1,14 @@ -from vgg import * -from dpn import * -from lenet import * -from senet import * -from pnasnet import * -from densenet import * -from googlenet import * -from shufflenet import * -from shufflenetv2 import * -from resnet import * -from resnext import * -from preact_resnet import * -from mobilenet import * -from mobilenetv2 import * +from .vgg import * +from .dpn import * +from .lenet import * +from .senet import * +from .pnasnet import * +from .densenet import * +from .googlenet import * +from .shufflenet import * +from .shufflenetv2 import * +from .resnet import * +from .resnext import * +from .preact_resnet import * +from .mobilenet import * +from .mobilenetv2 import *