def __init__(self, args, optimizer, name, extra_msg=''):
self.iteration = 0
self.sum_loss = 0
self.sum_acc = 0
self.sum_mean_diff = 0
self.sum_max_diff = 0
self.current_section = ''
self.optimizer = optimizer
# setup according to arguments
self.name = name if name is not '' else 'signdist'
self.out_file = "{}_{}".format(date.isoformat(date.today()), self.name)
self.log_file = "{}.log".format(self.out_file)
# write config to head of the log file
self.write_config(args, extra_msg)
评论列表
文章目录