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

trainer: fix for ctrl-c in --no-preview mode

Browse source
This commit is contained in:
iperov 2019-04-05 21:50:46 +04:00
parent fe40a46933
commit cfae9f9fd8
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
mainscripts/Trainer.py
View file

@ -185,7 +185,10 @@ def main(args, device_args):
op = input.get('op','')
if op == 'close':
break
io.process_messages(0.1)
try:
io.process_messages(0.1)
except KeyboardInterrupt:
s2c.put ( {'op': 'close'} )
else:
wnd_name = "Training preview"
io.named_window(wnd_name)