Fix ruotianluo#1

linzhlalala · Oct 15, 2018 · f793097 · f793097
1 parent ab5f364
commit f793097
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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
+    return model