import datetime from TTS.utils.io import AttrDict tcolors = AttrDict({ 'OKBLUE': '\033[94m', 'HEADER': '\033[95m', 'OKGREEN': '\033[92m', 'WARNING': '\033[93m', 'FAIL': '\033[91m', 'ENDC': '\033[0m', 'BOLD': '\033[1m', 'UNDERLINE': '\033[4m' }) class ConsoleLogger(): def __init__(self): # TODO: color code for value changes # use these to compare values between iterations self.old_train_loss_dict = None self.old_epoch_loss_dict = None self.old_eval_loss_dict = None # pylint: disable=no-self-use def get_time(self): now = datetime.datetime.now() return now.strftime("%Y-%m-%d %H:%M:%S") def print_epoch_start(self, epoch, max_epoch): print("\n{}{} > EPOCH: {}/{}{}".format(tcolors.UNDERLINE, tcolors.BOLD, epoch, max_epoch, tcolors.ENDC),
import datetime from TTS.utils.io import AttrDict tcolors = AttrDict({ "OKBLUE": "\033[94m", "HEADER": "\033[95m", "OKGREEN": "\033[92m", "WARNING": "\033[93m", "FAIL": "\033[91m", "ENDC": "\033[0m", "BOLD": "\033[1m", "UNDERLINE": "\033[4m", }) class ConsoleLogger: def __init__(self): # TODO: color code for value changes # use these to compare values between iterations self.old_train_loss_dict = None self.old_epoch_loss_dict = None self.old_eval_loss_dict = None # pylint: disable=no-self-use def get_time(self): now = datetime.datetime.now() return now.strftime("%Y-%m-%d %H:%M:%S") def print_epoch_start(self, epoch, max_epoch): print(