Skip to content

Commit e4d4811

Browse files
committed
Move all models from misc to models folder.
1 parent 7f4f268 commit e4d4811

File tree

6 files changed

+6
-6
lines changed

6 files changed

+6
-6
lines changed
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.
File renamed without changes.

models.py renamed to models/__init__.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,11 +5,11 @@
55
import misc.utils as utils
66
import torch
77

8-
from misc.ShowTellModel import ShowTellModel
9-
from misc.FCModel import FCModel
10-
from misc.CaptionModel import ShowAttendTellModel, AllImgModel
11-
from misc.Att2inModel import Att2inModel
12-
from misc.AttModel import *
8+
from .ShowTellModel import ShowTellModel
9+
from .FCModel import FCModel
10+
from .CaptionModel import ShowAttendTellModel, AllImgModel
11+
from .Att2inModel import Att2inModel
12+
from .AttModel import *
1313

1414
def setup(opt):
1515

@@ -48,4 +48,4 @@ def setup(opt):
4848
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
4949
model.load_state_dict(torch.load(os.path.join(opt.start_from, 'model.pth')))
5050

51-
return model
51+
return model

0 commit comments

Comments
 (0)