diff --git a/models/Model_AMP/Model.py b/models/Model_AMP/Model.py
index 04cb63a..151da7b 100644
--- a/models/Model_AMP/Model.py
+++ b/models/Model_AMP/Model.py
@@ -933,8 +933,10 @@ class AMPModel(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['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'])
 
         result = []
 
diff --git a/models/Model_AMP/config_schema.json b/models/Model_AMP/config_schema.json
index 96ec062..08e8330 100644
--- a/models/Model_AMP/config_schema.json
+++ b/models/Model_AMP/config_schema.json
@@ -179,6 +179,11 @@
         "random_dst_flip": {
           "type": "boolean"
         },
+        "force_preview_samples_num": {
+          "type": "integer",
+          "minimum": 1,
+          "maximum": 64
+        },
         "batch_size": {
           "type": "integer",
           "minimum": 1
diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py
index 375eee7..a464483 100644
--- a/models/Model_SAEHD/Model.py
+++ b/models/Model_SAEHD/Model.py
@@ -979,10 +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] )]
 
-        if self.options['preview_samples'] is None:
+        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['preview_samples'])
+            n_samples = min(self.get_batch_size(), self.options['force_preview_samples_num'])
 
         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 6f8d262..6af3c28 100644
--- a/models/Model_SAEHD/config_schema.json
+++ b/models/Model_SAEHD/config_schema.json
@@ -160,7 +160,7 @@
         "pretrain": {
           "type": "boolean"
         },
-        "preview_samples": {
+        "force_preview_samples_num": {
           "type": "integer",
           "minimum": 1,
           "maximum": 64