diff --git a/models/__init__.py b/models/__init__.py index 8d329bff..91e1faa8 100644 --- a/models/__init__.py +++ b/models/__init__.py @@ -12,7 +12,6 @@ from .ShowTellModel import ShowTellModel from .FCModel import FCModel from .OldModel import ShowAttendTellModel, AllImgModel -from .Att2inModel import Att2inModel from .AttModel import * from .TransformerModel import TransformerModel @@ -58,4 +57,4 @@ def setup(opt): assert os.path.isfile(os.path.join(opt.start_from,"infos_"+opt.id+".pkl")),"infos.pkl file does not exist in path %s"%opt.start_from model.load_state_dict(torch.load(os.path.join(opt.start_from, 'model.pth'))) - return model \ No newline at end of file + return model