From 48eb39442a403ac8f1a66b7e75d954efb4668b41 Mon Sep 17 00:00:00 2001 From: Lackey Date: Sat, 23 Feb 2019 21:18:53 -0500 Subject: [PATCH] Added timestamp to epoch output to see how long you've been training. --- models/ModelBase.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/models/ModelBase.py b/models/ModelBase.py index aff68a4..4160f07 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -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):