diff --git a/models/Model_SAE/Model.py b/models/Model_SAE/Model.py index 97bab15..afc5c6b 100644 --- a/models/Model_SAE/Model.py +++ b/models/Model_SAE/Model.py @@ -93,7 +93,7 @@ class SAEModel(ModelBase): self.options['bg_style_power'] = self.options.get('bg_style_power', default_bg_style_power) if is_first_run: - self.options['pretrain'] = io.input_bool ("Pretrain the model? (y/n, ?:help skip:n) : ", False, help_message="Pretrain the model with large amount of various faces. This technique may help to train the fake with overly different face shapes and light conditions of src/dst data. Face will be look more like a morphed. To reduce the morph effect, some model files will be initialized but not be updated after pretrain: LIAE: inter_AB.h5 DF: both decoders.h5. The longer you pretrain the model the more morphed face will look. After that, save and run the training again.") + self.options['pretrain'] = io.input_bool ("Pretrain the model? (y/n, ?:help skip:n) : ", False, help_message="Pretrain the model with large amount of various faces. This technique may help to train the fake with overly different face shapes and light conditions of src/dst data. Face will be look more like a morphed. To reduce the morph effect, some model files will be initialized but not be updated after pretrain: LIAE: inter_AB.h5 DF: encoder.h5. The longer you pretrain the model the more morphed face will look. After that, save and run the training again.") else: self.options['pretrain'] = False