CalvinBackup/iperov-DeepFaceLab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
1 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
mainscripts/Trainer.py
View File

@@ -277,7 +277,7 @@ def previewThread (input_queue, output_queue):
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")
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.start()
if not disable_preview:
previewThread (input_queue, output_queue)
previewThread (input_queue, output_queue)