CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix options in ModelBase.py

Browse Source
This commit is contained in:
iperov
2019-04-21 21:08:45 +04:00
parent 63beb3afd2
commit 093ca4dba6
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
models/ModelBase.py
+4 -4
View File
@@ -117,12 +117,12 @@ class ModelBase(object):
else: else:
self.options['src_scale_mod'] = self.options.get('src_scale_mod', 0) self.options['src_scale_mod'] = self.options.get('src_scale_mod', 0)
self.write_preview_history = self.options['write_preview_history'] self.write_preview_history = self.options.get('write_preview_history', False)
if not self.options['write_preview_history']: if not self.write_preview_history and 'write_preview_history' in self.options:
self.options.pop('write_preview_history') self.options.pop('write_preview_history')
self.target_iter = self.options['target_iter'] self.target_iter = self.options.get('target_iter',0)
if self.options['target_iter'] == 0: if self.target_iter == 0 and 'target_iter' in self.options:
self.options.pop('target_iter') self.options.pop('target_iter')
self.batch_size = self.options.get('batch_size',0) self.batch_size = self.options.get('batch_size',0)