diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index c3d445c..8f7f93a 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -53,7 +53,7 @@ class SAEModel(ModelBase): if is_first_run or ask_override: default_pixel_loss = False if is_first_run else self.options.get('pixel_loss', False) - self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", default_pixel_loss, help_message="Default perceptual (DSSIM) loss good for initial understanding structure of faces. Use pixel loss after 20-30k epochs to enhance fine details.") + self.options['pixel_loss'] = input_bool ("Use pixel loss? (y/n, ?:help skip: n/default ) : ", default_pixel_loss, help_message="Default perceptual (DSSIM) loss good for initial understanding structure of faces. Use pixel loss after 30-40k epochs to enhance fine details.") else: self.options['pixel_loss'] = self.options.get('pixel_loss', False)