这段代码很简单
ifname==“main"：# pass from torch.utils.tensorboard import SummaryWriter

model = get_net(False)
input_ = torch.randn((1, 3, 256, 256))
gt_ = torch.rand((1, 2, 256, 256))
# 添加模型结构到TensorBoard
with SummaryWriter(comment="YOLOP") as w:
    w.add_graph(model, (input_,))

字符串
当我运行这个，它无法显示网络图，并作出如下错误：

[[ 3.6462e-02,  1.6451e-04,  2.8460e-02, -2.8415e-03, -3.7653e+00,
             4.1301e+00],
           [ 3.6462e-02,  1.6451e-04,  2.8460e-02, -2.8415e-03, -3.7653e+00,
             4.1301e+00],
           [ 3.6462e-02,  1.6451e-04,  2.8460e-02, -2.8415e-03, -3.7653e+00,
             4.1301e+00],
           ...,
           [ 3.6462e-02,  1.6451e-04,  2.8460e-02, -2.8415e-03, -3.7653e+00,
             4.1301e+00],
           [ 3.6462e-02,  1.6451e-04,  2.8460e-02, -2.8415e-03, -3.7653e+00,
             4.1301e+00],
           [ 3.6462e-02,  1.6451e-04,  2.8460e-02, -2.8415e-03, -3.7653e+00,
             4.1301e+00]]]]], grad_fn=<CloneBackward0>)]), tensor([[[[0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          ...,
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000]],

         [[0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          ...,
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000]]]],
       grad_fn=<SigmoidBackward0>), tensor([[[[0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          ...,
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000]],

         [[0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          ...,
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000],
          [0.5000, 0.5000, 0.5000,  ..., 0.5000, 0.5000, 0.5000]]]],
       grad_fn=<SigmoidBackward0>)]
:List inputs to traced functions must have consistent element type. Found Tuple[Tensor, List[Tensor]] and Tensor

型
我真的很想知道如何解决这个问题，我很伤心！每次我运行这个代码，我看不到任何东西，tensorboard没有显示任何图形，这让我困惑，请任何人谁可以保存我。