From caf59f2e130f3737697e3aa59789ae82749389d5 Mon Sep 17 00:00:00 2001 From: JanFschr Date: Tue, 23 Nov 2021 11:02:35 +0100 Subject: [PATCH] fix missing / wrong use_fp16 --- models/Model_SAEHD/Model.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/models/Model_SAEHD/Model.py b/models/Model_SAEHD/Model.py index e922705..4294c30 100644 --- a/models/Model_SAEHD/Model.py +++ b/models/Model_SAEHD/Model.py @@ -68,6 +68,7 @@ class SAEHDModel(ModelBase): default_random_color = self.options['random_color'] = self.load_or_def_option('random_color', False) default_clipgrad = self.options['clipgrad'] = self.load_or_def_option('clipgrad', False) default_pretrain = self.options['pretrain'] = self.load_or_def_option('pretrain', False) + default_use_fp16 = self.options['use_fp16'] = self.load_or_def_option('use_fp16', False) ask_override = self.ask_override() if self.is_first_run() or ask_override: @@ -253,10 +254,6 @@ Examples: df, liae, df-d, df-ud, liae-ud, ... if self.is_exporting: use_fp16 = io.input_bool ("Export quantized?", False, help_message='Makes the exported model faster. If you have problems, disable this option.') - use_fp16 = False - if self.is_exporting: - use_fp16 = io.input_bool ("Export quantized?", False, help_message='Makes the exported model faster. If you have problems, disable this option.') - self.gan_power = gan_power = 0.0 if self.pretrain else self.options['gan_power'] random_warp = False if self.pretrain else self.options['random_warp'] random_src_flip = self.random_src_flip if not self.pretrain else True