diff --git a/core/interact/no_interact_dict.py b/core/interact/no_interact_dict.py index e29d451..3266fc8 100644 --- a/core/interact/no_interact_dict.py +++ b/core/interact/no_interact_dict.py @@ -3,6 +3,7 @@ import os # Sample dictionary with default answers dictionary = { 'Extractor_continue_deleting_files' : '\n', + 'target_iter_increment' : '10000', 'choose_model_name' : '0', 'choose_mode' : '1', 'choose_mask_mode' : '4', diff --git a/models/ModelBase.py b/models/ModelBase.py index 93c378f..35e00ca 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -296,8 +296,8 @@ class ModelBase(object): def ask_target_iter(self, default_value=0): if io.get_default_answer('NonInteractiveMode') is not None: - default_target_iter = self.load_or_def_option('target_iter', default_value + 10000) - self.options['target_iter'] = max(0, default_target_iter + 10000) + default_target_iter = self.load_or_def_option('target_iter', default_value + io.get_default_answer('target_iter_increment')) + self.options['target_iter'] = max(0, default_target_iter + io.get_default_answer('target_iter_increment')) print("TARGET ITERATION: " + str(self.options['target_iter'])) else: default_target_iter = self.load_or_def_option('target_iter', default_value)