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

Update Trainer.py

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

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