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

fix last PR

Browse source
This commit is contained in:
iperov 2019-01-01 19:42:19 +04:00
parent c707fc0658
commit 5c0c79e528
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
mainscripts/Trainer.py
View file

@ -277,7 +277,7 @@ def previewThread (input_queue, output_queue):
cv2.destroyAllWindows() cv2.destroyAllWindows()
def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, preview, **in_options): def main (training_data_src_dir, training_data_dst_dir, model_path, model_name, disable_preview, **in_options):
print ("Running trainer.\r\n") print ("Running trainer.\r\n")
output_queue = queue.Queue() output_queue = queue.Queue()
@ -286,5 +286,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 = 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() thread.start()
if preview: if not disable_preview:
previewThread (input_queue, output_queue) previewThread (input_queue, output_queue)