github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-19 13:09:56 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Bug fix

Browse source
This commit is contained in:
Cioscos 2021-12-02 19:07:22 +01:00
commit 57cdf0e794
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/ModelBase.py
View file

@ -211,7 +211,8 @@ class ModelBase(object):
# save as default options only for first run model initialize
self.default_options_path.write_bytes( pickle.dumps (self.options) )
# save config file
self.save_config_file()
if config_training_file is not None:
self.save_config_file()
self.session_name = self.options.get('session_name', "")
self.autobackup_hour = self.options.get('autobackup_hour', 0)

4
models/Model_SAEHD/Model.py
View file

@ -28,7 +28,7 @@ class SAEHDModel(ModelBase):
min_res = 64
max_res = 640
#default_usefp16 = self.options['use_fp16'] = self.load_or_def_option('use_fp16', False)
default_usefp16 = self.options['use_fp16'] = self.load_or_def_option('use_fp16', False)
default_resolution = self.options['resolution'] = self.load_or_def_option('resolution', 128)
default_face_type = self.options['face_type'] = self.load_or_def_option('face_type', 'f')
default_models_opt_on_gpu = self.options['models_opt_on_gpu'] = self.load_or_def_option('models_opt_on_gpu', True)
@ -82,7 +82,7 @@ class SAEHDModel(ModelBase):
self.ask_random_src_flip()
self.ask_random_dst_flip()
self.ask_batch_size(suggest_batch_size)
#self.options['use_fp16'] = io.input_bool ("Use fp16", default_usefp16, help_message='Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.')
self.options['use_fp16'] = io.input_bool ("Use fp16", default_usefp16, help_message='Increases training/inference speed, reduces model size. Model may crash. Enable it after 1-5k iters.')
if self.is_first_run():
if (self.read_from_conf and not self.config_file_exists) or not self.read_from_conf: