From f669591b96a02a2e2d20763c877c52dddc270e19 Mon Sep 17 00:00:00 2001 From: seranus Date: Thu, 9 Dec 2021 18:32:01 +0100 Subject: [PATCH] saehd - preview samples fix --- models/Model_SAEHD/Model.py | 7 ++----- models/Model_SAEHD/config_schema.json | 2 +- 2 files changed, 3 insertions(+), 6 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index a464483..3f7d4dd 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -83,7 +83,7 @@ class SAEHDModel(ModelBase): self.ask_autobackup_hour() self.ask_maximum_n_backups() self.ask_write_preview_history() - self.options['preview_samples'] = np.clip ( io.input_int ("Number of samples to preview", default_cpu_cap, add_info="1 - 16", help_message="Typical fine value is 4"), 1, 16 ) + self.options['preview_samples'] = np.clip ( io.input_int ("Number of samples to preview", default_preview_samples, add_info="1 - 16", help_message="Typical fine value is 4"), 1, 16 ) self.ask_target_iter() self.ask_retraining_samples() self.ask_random_src_flip() @@ -979,10 +979,7 @@ class SAEHDModel(ModelBase): target_srcm, target_dstm = [ nn.to_data_format(x,"NHWC", self.model_data_format) for x in ([target_srcm, target_dstm] )] - if self.options['force_preview_samples_num'] is None: - n_samples = min(4, self.get_batch_size(), 800 // self.resolution ) - else: - n_samples = min(self.get_batch_size(), self.options['force_preview_samples_num']) + n_samples = min(self.get_batch_size(), self.options['preview_samples']) if filenames is not None and len(filenames) > 0: for i in range(n_samples): diff --git a/models/Model_SAEHD/config_schema.json b/models/Model_SAEHD/config_schema.json index 6af3c28..6f8d262 100644 --- a/models/Model_SAEHD/config_schema.json +++ b/models/Model_SAEHD/config_schema.json @@ -160,7 +160,7 @@ "pretrain": { "type": "boolean" }, - "force_preview_samples_num": { + "preview_samples": { "type": "integer", "minimum": 1, "maximum": 64