added finetune mode

configs/dataset/tvsum.yml

@@ -7,7 +7,7 @@ max_v_l: 1000
 seed: 2018
 lr: 0.001
 lr_drop: 2000
-n_epoch: 2000
+n_epoch: 1000
 bsz: 4
 domains:
   - BK

training/config.py

@@ -26,10 +26,11 @@
 from easydict import EasyDict
 
 class BaseOptions(object):
-    def __init__(self, model, dataset, feature):
+    def __init__(self, model, dataset, feature, resume):
         self.model = model
         self.dataset = dataset
         self.feature = feature
+        self.resume = resume
         self.opt = {}
 
     @property
@@ -55,7 +56,11 @@ def parse(self):
         self.opt = EasyDict(self.opt)
 
         # result directory
-        self.opt.results_dir = os.path.join(self.opt.results_dir, self.model, self.dataset, self.feature)
+        if self.resume:
+            self.opt.results_dir = os.path.join(self.opt.results_dir, self.model, f"{self.dataset}_finetune", self.feature)
+        else:
+            self.opt.results_dir = os.path.join(self.opt.results_dir, self.model, self.dataset, self.feature)
+
         self.opt.ckpt_filepath = os.path.join(self.opt.results_dir, self.opt.ckpt_filename)
         self.opt.train_log_filepath = os.path.join(self.opt.results_dir, self.opt.train_log_filename)
         self.opt.eval_log_filepath = os.path.join(self.opt.results_dir, self.opt.eval_log_filename)

training/train.py

@@ -289,7 +289,7 @@ def check_valid_combination(dataset, feature):
     is_valid = check_valid_combination(args.dataset, args.feature)
 
     if is_valid:
-        option_manager = BaseOptions(args.model, args.dataset, args.feature)
+        option_manager = BaseOptions(args.model, args.dataset, args.feature, args.resume)
         option_manager.parse()
         option_manager.clean_and_makedirs()