我在使用tensorboardX可视化模型的权重和偏差时遇到了麻烦。这是我的模型(无论如何都很简单):
self.pipe = nn.Sequential(nn.Linear(9,128), …
发布的代码段不足以导致问题。
在writer上调用close()方法时,成员变量file_writer设置为None。请检查是否在writer上调用了close()方法。当writer对象用作Context Manager并退出with块时,也会调用close()方法。
with SummaryWriter() as writer: writer.add_scalar... writer.add_histogram # this will cause a crash