首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >“NoneType”对象没有属性“add_summary”

“NoneType”对象没有属性“add_summary”
EN

Stack Overflow用户
提问于 2019-03-01 11:31:52
回答 1查看 1.1K关注 0票数 0

我在使用tensorboardX可视化我的模型的权重和偏差时遇到了麻烦。这是我的模型(无论如何都很简单):

代码语言:javascript
复制
    self.pipe = nn.Sequential(nn.Linear(9, 128),
                             nn.ReLU(),
                             nn.Linear(128, 256),
                             nn.ReLU(),
                             nn.Linear(256,2),
                             nn.Softmax()
                             )

     def forward(self, x):
     return self.pipe(x)

这就是我在pytorch中出错的地方

代码语言:javascript
复制
for name, param in net.named_parameters():
    writer.add_histogram(name, param, epoch_size, bins='auto')

错误是

代码语言:javascript
复制
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-70-d060d2df4423> in <module>()
      1 for name, param in net.named_parameters():
----> 2     writer.add_histogram(name, param, epoch_size, bins='auto')

~\Anaconda3\lib\site-packages\tensorboardX\writer.py in add_histogram(self, tag, values, global_step, bins, walltime)
    403         if isinstance(bins, six.string_types) and bins == 'tensorflow':
    404             bins = self.default_bins
--> 405         self.file_writer.add_summary(
    406             histogram(tag, values, bins), global_step, walltime)
    407 

AttributeError: 'NoneType' object has no attribute 'add_summary'

但我真的必须看到权重处于次优状态的直方图。所以我稍微修改了一下代码,一步一步地进行。

代码语言:javascript
复制
param = np.array(list(net.parameters()))   
print(param[0].data)
writer.add_histogram('weight', param[0].data)

轰隆隆!仍然是同样的错误,也许这根本不会改变。

EN

回答 1

Stack Overflow用户

发布于 2019-03-01 16:42:26

发布的代码片段不足以成为问题的根本原因。

在编写器上调用close()方法时,成员变量file_writer设置为None。请检查是否在编写器上调用了close()方法。当编写器对象用作上下文管理器并退出with块时,也会调用close()方法。

代码语言:javascript
复制
with SummaryWriter() as writer:
   writer.add_scalar...

writer.add_histogram # this will cause a crash
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/54937532

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档