github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-08-22 06:23: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 import cv2
from samples import SampleGeneratorBase from samples import SampleGeneratorBase
from nnlib import nnlib from nnlib import nnlib
from datetime import datetime
''' '''
You can implement your own model. Check examples. You can implement your own model. Check examples.
''' '''
@ -344,6 +346,9 @@ class ModelBase(object):
for (loss_name, loss_value) in losses: for (loss_name, loss_value) in losses:
loss_string += " %s:%.3f" % (loss_name, loss_value) 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 return loss_string
def pass_one_epoch(self): def pass_one_epoch(self):