diff --git a/models/Model_DF/Model.py b/models/Model_DF/Model.py index 252d333..9d7ac1b 100644 --- a/models/Model_DF/Model.py +++ b/models/Model_DF/Model.py @@ -15,7 +15,8 @@ class Model(ModelBase): #override def onInitializeOptions(self, is_first_run, ask_override): if is_first_run or ask_override: - self.options['pixel_loss'] = self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", False, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") + def_pixel_loss = self.options.get('pixel_loss', False) + self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") else: self.options['pixel_loss'] = self.options.get('pixel_loss', False) diff --git a/models/Model_H128/Model.py b/models/Model_H128/Model.py index 4dcbb64..98dcf6b 100644 --- a/models/Model_H128/Model.py +++ b/models/Model_H128/Model.py @@ -23,7 +23,8 @@ class Model(ModelBase): self.options['lighter_ae'] = self.options.get('lighter_ae', default_lighter_ae) if is_first_run or ask_override: - self.options['pixel_loss'] = self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", False, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") + def_pixel_loss = self.options.get('pixel_loss', False) + self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") else: self.options['pixel_loss'] = self.options.get('pixel_loss', False) diff --git a/models/Model_H64/Model.py b/models/Model_H64/Model.py index a7ea415..c2bb60b 100644 --- a/models/Model_H64/Model.py +++ b/models/Model_H64/Model.py @@ -23,7 +23,8 @@ class Model(ModelBase): self.options['lighter_ae'] = self.options.get('lighter_ae', default_lighter_ae) if is_first_run or ask_override: - self.options['pixel_loss'] = self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", False, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") + def_pixel_loss = self.options.get('pixel_loss', False) + self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") else: self.options['pixel_loss'] = self.options.get('pixel_loss', False) diff --git a/models/Model_LIAEF128/Model.py b/models/Model_LIAEF128/Model.py index 19f26e0..f885f9b 100644 --- a/models/Model_LIAEF128/Model.py +++ b/models/Model_LIAEF128/Model.py @@ -16,7 +16,8 @@ class Model(ModelBase): #override def onInitializeOptions(self, is_first_run, ask_override): if is_first_run or ask_override: - self.options['pixel_loss'] = self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", False, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") + def_pixel_loss = self.options.get('pixel_loss', False) + self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", def_pixel_loss, help_message="Default DSSIM loss good for initial understanding structure of faces. Use pixel loss after 20k epochs to enhance fine details and remove face jitter.") else: self.options['pixel_loss'] = self.options.get('pixel_loss', False)