def setup_logger(self):
    try:
        from torch.utils.tensorboard import SummaryWriter
    except:
        from utilities.summary_writer import SummaryWriter
    self.logger = SummaryWriter(log_dir=self.opts.savedir, comment='Training and Validation logs')
def setup_logger(self):
    try:
        from torch.utils.tensorboard import SummaryWriter
    except:
        from utilities.summary_writer import SummaryWriter
    self.logger = SummaryWriter(log_dir=self.opts.savedir, comment='Training and Validation logs')