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

Added timestamp to epoch output to see how long you've been training.

Browse source
This commit is contained in:
Lackey 2019-02-23 21:18:53 -05:00
commit 48eb39442a
1 changed files with 5 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
models/ModelBase.py
View file

@ -13,6 +13,8 @@ import numpy as np
import cv2
from samples import SampleGeneratorBase
from nnlib import nnlib
from datetime import datetime
'''
You can implement your own model. Check examples.
'''
@ -344,6 +346,9 @@ class ModelBase(object):
for (loss_name, loss_value) in losses:
loss_string += " %s:%.3f" % (loss_name, loss_value)
date_string = datetime.now().strftime(" [ T: %d/%b/%y, %H:%M:%S ]")
loss_string += date_string
return loss_string
def pass_one_epoch(self):