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

no_preview for model

Browse source
This commit is contained in:
Colombo 2020-01-04 23:52:12 +04:00
parent 2429d28737
commit b5c234dac3
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
mainscripts/Trainer.py
View file

@ -28,6 +28,7 @@ def trainerThread (s2c, c2s, e, args, device_args):
save_interval_min = 15
debug = args.get('debug', '')
execute_programs = args.get('execute_programs', [])
no_preview = args.get('no_preview', False)
if not training_data_src_path.exists():
io.log_err('Training data src directory does not exist.')
@ -47,6 +48,7 @@ def trainerThread (s2c, c2s, e, args, device_args):
pretraining_data_path=pretraining_data_path,
is_training=True,
debug=debug,
no_preview=no_preview,
device_args=device_args)
is_reached_goal = model.is_reached_iter_goal()

3
models/ModelBase.py
View file

@ -23,7 +23,7 @@ You can implement your own model. Check examples.
class ModelBase(object):
def __init__(self, model_path, training_data_src_path=None, training_data_dst_path=None, pretraining_data_path=None, is_training=False, debug = False, device_args = None,
def __init__(self, model_path, training_data_src_path=None, training_data_dst_path=None, pretraining_data_path=None, is_training=False, debug = False, no_preview=False, device_args = None,
ask_enable_autobackup=True,
ask_write_preview_history=True,
ask_target_iter=True,
@ -55,6 +55,7 @@ class ModelBase(object):
self.pretraining_data_path = pretraining_data_path
self.debug = debug
self.no_preview = no_preview
self.is_training_mode = is_training
self.iter = 0