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):