Method __init__

tu/loggers/visualizer.py:15–26

(self, opt)

class HTMLVisualizer(BaseHTMLHelper):
    def __init__(self, opt):

        self.vis = HTMLTableVisualizer(visdir=os.path.join(opt.checkpoints_dir, opt.name, 'web'),
                                       title=f"{opt.name}_{datetime.datetime.now().strftime('%Y_%m%d_%H%M_%S')}",
                                       persist_row_counter=False)  # rewritten each time so not persisting row counts

        # create a logging file to store training losses
        os.makedirs(os.path.join(opt.checkpoints_dir, opt.name), exist_ok=True)
        self.log_name = os.path.join(opt.checkpoints_dir, opt.name, 'loss_log.txt')
        # with open(self.log_name, "a") as log_file:
        #     now = time.strftime("%c")
        #     log_file.write('================ Training Loss (%s) ================\n' % now)

    def reset(self):
        """Reset the self.saved status"""