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

exit removed

Browse source
This commit is contained in:
leno3003 2022-04-21 13:34:25 +02:00
parent 224454be12
commit 271ee47877
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

@ -86,7 +86,7 @@ def trainerThread (s2c, c2s, e,
io.log_info('Model already trained to target iteration. You can use preview.') io.log_info('Model already trained to target iteration. You can use preview.')
if io.get_default_answer('NonInteractiveMode') is not None: if io.get_default_answer('NonInteractiveMode') is not None:
model_save() model_save()
os._exit(0) s2c.put ( {'op': 'close'} )
else: else:
io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) ) io.log_info('Starting. Target iteration: %d. Press "Enter" to stop training and save model.' % ( model.get_target_iter() ) )
else: else:
@ -160,6 +160,7 @@ def trainerThread (s2c, c2s, e,
if model.get_iter() == 1: if model.get_iter() == 1:
model_save() model_save()
s2c.put ( {'op': 'close'} )
if model.get_target_iter() != 0 and model.is_reached_iter_goal(): if model.get_target_iter() != 0 and model.is_reached_iter_goal():
io.log_info ('Reached target iteration.') io.log_info ('Reached target iteration.')
@ -168,7 +169,6 @@ def trainerThread (s2c, c2s, e,
io.log_info ('You can use preview now.') io.log_info ('You can use preview now.')
if io.get_default_answer('NonInteractiveMode') is not None: if io.get_default_answer('NonInteractiveMode') is not None:
model_save() model_save()
os._exit(0)
need_save = False need_save = False
while time.time() - last_save_time >= save_interval_min*60: while time.time() - last_save_time >= save_interval_min*60:
last_save_time += save_interval_min*60 last_save_time += save_interval_min*60