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

Add disable preview option with --disable-preview

Browse source
This commit is contained in:
Luke Barr 2019-02-18 10:36:33 -06:00
commit 3e175ce1d3
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
main.py
View file

@ -87,6 +87,7 @@ if __name__ == "__main__":
model_path=arguments.model_dir,
model_name=arguments.model_name,
debug = arguments.debug,
disable_preview = arguments.disable_preview,
#**options
force_gpu_idx = arguments.force_gpu_idx,
cpu_only = arguments.cpu_only
@ -98,6 +99,7 @@ if __name__ == "__main__":
train_parser.add_argument('--model-dir', required=True, action=fixPathAction, dest="model_dir", help="Model dir.")
train_parser.add_argument('--model', required=True, dest="model_name", choices=Path_utils.get_all_dir_names_startswith ( Path(__file__).parent / 'models' , 'Model_'), help="Type of model")
train_parser.add_argument('--debug', action="store_true", dest="debug", default=False, help="Debug samples.")
train_parser.add_argument('--disable-preview', action="store_true", dest="disable_preview", default=False, help="Disable the preview")
train_parser.add_argument('--cpu-only', action="store_true", dest="cpu_only", default=False, help="Train on CPU.")
train_parser.add_argument('--force-gpu-idx', type=int, dest="force_gpu_idx", default=-1, help="Force to choose this GPU idx.")

1
mainscripts/Trainer.py
View file

@ -242,4 +242,5 @@ def main (training_data_src_dir, training_data_dst_dir, model_path, model_name,
thread = threading.Thread(target=trainerThread, args=(output_queue, input_queue, training_data_src_dir, training_data_dst_dir, model_path, model_name), kwargs=in_options )
thread.start()
if in_options.get('disable_preview', False) == False:
previewThread (input_queue, output_queue)