tensorboardX add_graph 无法显示网络图

貌似tensorboardX更新完之后，调用add_graph后在浏览器中无法正常显示网络图，只有两个空白框，如下图。当我天真地以为这是tensorboardX的bug，后续的更新应该会修复这个bug，但是持续几个月的等待，貌似tensorboardX一直存在这个问题，于是就有了以下的替代方案。

 

将from tensorboardX import SummaryWriter
替换成from torch.utils.tensorboard import SummaryWriter

前提是PyTorch>=1.1.0

可以无痛替换，相关的代码无需修改，再次运行就可以看到网络图了。