diff --git a/models/builder.py b/models/builder.py index 4c2c2f8..767653d 100644 --- a/models/builder.py +++ b/models/builder.py @@ -7,7 +7,7 @@ from torch.autograd import Variable from .heads import Corr_Up, MultiRPN, DepthwiseRPN from .backbones import AlexNet, Vgg, ResNet22, Incep22, ResNeXt22, ResNet22W, resnet50, resnet34, resnet18 -from neck import AdjustLayer, AdjustAllLayer +from .neck import AdjustLayer, AdjustAllLayer from .utils import load_pretrain