Thesis/ADNC
1
0
Fork 0
mirror of https://github.com/JoergFranke/ADNC.git synced 2024-11-17 13:58:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix color code

Browse Source
This commit is contained in:
Joerg Franke 2018-12-28 20:35:57 +01:00
parent d89c49d251
commit b2ae8de21e
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
adnc/model/supporter.py
View File

@ -27,7 +27,7 @@ The supporter class creates for each training run a folder, logs the prints in a
gradients and losses. gradients and losses.
""" """
class color_code: class ColorCode:
def __init__(self): def __init__(self):
self.bold = '\033[1m' self.bold = '\033[1m'
self.underline = '\033[4m' self.underline = '\033[4m'
@ -86,7 +86,7 @@ class Supporter():
else: else:
raise UserWarning('supporter: no config found') raise UserWarning('supporter: no config found')
self.pub('load config file ' + color_code.darkcyan + '{}'.format(config_file) + color_code.end) self.pub('load config file ' + ColorCode().darkcyan + '{}'.format(config_file) + ColorCode().end)
with open(config_file, 'r') as f: with open(config_file, 'r') as f:
self.configs = yaml.load(f) self.configs = yaml.load(f)
@ -97,7 +97,7 @@ class Supporter():
if not task_id: if not task_id:
self.pub('### task id: {}'.format(task_id)) self.pub('### task id: {}'.format(task_id))
self.pub('##############################################') self.pub('##############################################')
self.pub(color_code.bold + 'SUPPORTER LOG' + color_code.end) self.pub(ColorCode().bold + 'SUPPORTER LOG' + ColorCode().end)
self.pub("model : {}".format(self.model_type)) self.pub("model : {}".format(self.model_type))
self.pub("data set : {}".format(self.data_set)) self.pub("data set : {}".format(self.data_set))
self.pub("experiment name : {}".format(self.experiment_name)) self.pub("experiment name : {}".format(self.experiment_name))
@ -106,13 +106,13 @@ class Supporter():
self.pub("time : {}".format(time.strftime("%H:%M:%S"))) self.pub("time : {}".format(time.strftime("%H:%M:%S")))
self.pub("host : {}".format(self.host_name)) self.pub("host : {}".format(self.host_name))
self.pub('##############################################') self.pub('##############################################')
self.pub(color_code.bold + 'DATA CONFIG' + color_code.end) self.pub(ColorCode().bold + 'DATA CONFIG' + ColorCode().end)
self.pub(self.configs[self.data_set]) self.pub(self.configs[self.data_set])
self.pub('##############################################') self.pub('##############################################')
self.pub(color_code.bold + 'MODEL CONFIG' + color_code.end) self.pub(ColorCode().bold + 'MODEL CONFIG' + ColorCode().end)
self.pub(self.configs[self.model_type]) self.pub(self.configs[self.model_type])
self.pub('##############################################') self.pub('##############################################')
self.pub(color_code.bold + 'TRAIN CONFIG' + color_code.end) self.pub(ColorCode().bold + 'TRAIN CONFIG' + ColorCode().end)
self.pub(self.config('training')) self.pub(self.config('training'))
self.pub('##############################################') self.pub('##############################################')