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

after save loss string now shows averaged value since last save

Browse source
This commit is contained in:
iperov 2019-03-25 18:14:04 +04:00
parent bb02cc1d97
commit 131b2b5c79
2 changed files with 30 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

10
models/ModelBase.py
View file

@ -374,15 +374,7 @@ class ModelBase(object):
self.iter += 1
time_str = time.strftime("[%H:%M:%S]")
if iter_time >= 10:
loss_string = "{0}[#{1:06d}][{2:.5s}s]".format ( time_str, self.iter, '{:0.4f}'.format(iter_time) )
else:
loss_string = "{0}[#{1:06d}][{2:04d}ms]".format ( time_str, self.iter, int(iter_time*1000) )
for (loss_name, loss_value) in losses:
loss_string += " %s:%.3f" % (loss_name, loss_value)
return loss_string
return self.iter, iter_time
def pass_one_iter(self):
self.last_sample = self.generate_next_sample()