StarMirror/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2024-03-22 13:10:55 +08:00
Code Issues Projects Releases Wiki Activity

fix save in trainer

Browse Source
This commit is contained in:
iperov 2021-10-11 15:02:41 +04:00
parent 0a203bb25e
commit 91f0f30a78
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mainscripts/Trainer.py
View File

@ -164,8 +164,12 @@ def trainerThread (s2c, c2s, e,
is_reached_goal = True
io.log_info ('You can use preview now.')
if not is_reached_goal and (time.time() - last_save_time) >= save_interval_min*60:
need_save = False
while time.time() - last_save_time >= save_interval_min*60:
last_save_time += save_interval_min*60
need_save = True
if not is_reached_goal and need_save:
model_save()
send_preview()