diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index 8618d6f..b925cb1 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -162,7 +162,7 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... self.options['random_warp'] = io.input_bool ("Enable random warp of samples", default_random_warp, help_message="Random warp is required to generalize facial expressions of both faces. When the face is trained enough, you can disable it to get extra sharpness and reduce subpixel shake for less amount of iterations.") self.options['random_downsample'] = io.input_bool("Enable random downsample of samples", default_random_downsample, help_message="") - self.options['random_noise'] = io.input_bool("Enable random noise added to samples", False, help_message="") + self.options['random_noise'] = io.input_bool("Enable random noise added to samples", default_random_noise, help_message="") # self.options['random_blur'] = io.input_bool("Enable random blur of samples", False, help_message="") # self.options['random_jpeg'] = io.input_bool("Enable random jpeg compression of samples", False, help_message="")