From 3d078ff9604a1cfc3c1f2427b38354c4877f5541 Mon Sep 17 00:00:00 2001 From: leno3003 Date: Mon, 4 Apr 2022 10:08:53 +0200 Subject: [PATCH] target iter value --- core/interact/no_interact_dict.py | 1 + models/ModelBase.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) 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)