提交一个bug

[xiaobai-marker]

class LossHistory(): def init(self, log_dir, model, input_shape): self.log_dir = log_dir self.losses = [] self.val_loss = []

    os.makedirs(self.log_dir)
    self.writer     = SummaryWriter(self.log_dir)
    try:
        dummy_input     = torch.randn(2, 3, input_shape[0], input_shape[1])
        self.writer.add_graph(model, dummy_input)
    except:
        pass   这部分代码虽然加载了模型到tensorboard 但是不能正确显示