github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-16 10:03:41 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update Trainer.py

Browse source
This commit is contained in:
iperov 2019-01-01 18:09:27 +04:00 • committed by GitHub
parent a8694b73f0
commit c707fc0658
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
mainscripts/Trainer.py
View file

@ -277,8 +277,8 @@ 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, preview, **in_options):
print ("Running trainer(preview=%s).\r\n" % (preview)) print ("Running trainer.\r\n")
output_queue = queue.Queue() output_queue = queue.Queue()
input_queue = queue.Queue() input_queue = queue.Queue()
@ -287,4 +287,4 @@ def main (training_data_src_dir, training_data_dst_dir, model_path, model_name,p
thread.start() thread.start()
if preview: if preview:
previewThread (input_queue, output_queue) previewThread (input_queue, output_queue)