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

saehd - force full preview

Browse source
This commit is contained in:
seranus 2021-12-09 18:36:09 +01:00
commit bf783a43eb
2 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
models/Model_SAEHD/Model.py
View file

@ -74,6 +74,7 @@ class SAEHDModel(ModelBase):
default_pretrain = self.options['pretrain'] = self.load_or_def_option('pretrain', False) default_pretrain = self.options['pretrain'] = self.load_or_def_option('pretrain', False)
default_cpu_cap = self.options['cpu_cap'] = self.load_or_def_option('cpu_cap', 8) default_cpu_cap = self.options['cpu_cap'] = self.load_or_def_option('cpu_cap', 8)
default_preview_samples = self.options['preview_samples'] = self.load_or_def_option('preview_samples', 4) default_preview_samples = self.options['preview_samples'] = self.load_or_def_option('preview_samples', 4)
default_full_preview = self.options['force_full_preview'] = self.load_or_def_option('force_full_preview', False)
default_lr_modifier = self.options['lr_modifier'] = self.load_or_def_option('lr_modifier', 0) default_lr_modifier = self.options['lr_modifier'] = self.load_or_def_option('lr_modifier', 0)
ask_override = False if self.read_from_conf else self.ask_override() ask_override = False if self.read_from_conf else self.ask_override()
@ -986,7 +987,7 @@ class SAEHDModel(ModelBase):
S[i] = label_face_filename(S[i], filenames[0][i]) S[i] = label_face_filename(S[i], filenames[0][i])
D[i] = label_face_filename(D[i], filenames[1][i]) D[i] = label_face_filename(D[i], filenames[1][i])
if self.resolution <= 256 or self.options['preview_samples'] == True: if self.resolution <= 256 or self.options['force_full_preview'] == True:
result = [] result = []
st = [] st = []

2
models/Model_SAEHD/config_schema.json
View file

@ -165,7 +165,7 @@
"minimum": 1, "minimum": 1,
"maximum": 64 "maximum": 64
}, },
"force_complete_preview": { "force_full_preview": {
"type": "boolean" "type": "boolean"
}, },
"session_name": { "session_name": {