From e6b55e972c45db4a6568c60816f25ca23b58419d Mon Sep 17 00:00:00 2001
From: seranus <ognjen_j@yahoo.com>
Date: Thu, 9 Dec 2021 18:11:19 +0100
Subject: [PATCH] preview images update

---
 models/Model_SAEHD/Model.py           | 6 ++++--
 models/Model_SAEHD/config_schema.json | 2 +-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py
index 17a5de7..be6a9bb 100644
--- a/models/Model_SAEHD/Model.py
+++ b/models/Model_SAEHD/Model.py
@@ -979,8 +979,10 @@ class SAEHDModel(ModelBase):
 
         target_srcm, target_dstm = [ nn.to_data_format(x,"NHWC", self.model_data_format) for x in ([target_srcm, target_dstm] )]
 
-        #n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
-        n_samples = min(self.get_batch_size(), self.options['preview_samples'])
+        if self.options['preview_samples'] is None:
+            n_samples = min(4, self.get_batch_size(), 800 // self.resolution )
+        else:
+            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 9533f0c..f42cb93 100644
--- a/models/Model_SAEHD/config_schema.json
+++ b/models/Model_SAEHD/config_schema.json
@@ -163,7 +163,7 @@
         "preview_samples": {
           "type": "integer",
           "minimum": 1,
-          "maximum": 16
+          "maximum": 64
         },
         "session_name": {
           "type": "string"