pytorch:使用tensorboardX可视化网络模型时add_graph位置报错

1.报错信息

TypeError: graph() got an unexpected keyword argument ‘use_strict_trace’

提示 graph()这个函数多了一个参数’use_strict_trace’;
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第1张图片
也觉得应该是tensorboardX版本问题,但uninstall 再insatall之后也不行,用另一台机子也是同样的报错;

2.解决方式

既然说:graph()这个函数多给了这个实参,那就转到graph()看它的形参有哪些:

首先:点击报错位置链接的py文件,锁定报错代码位置
在这里插入图片描述
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第2张图片其次,查看graph()的定义,看是否有’use_strict_trace’这个形参
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第3张图片
发现,该函数没有该实参。

最后,删除调用graph()函数中的“use_strict_trace=use_strict_trace”。
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第4张图片

3.结果

运行成功
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第5张图片
执行 runs 文件下生成的图表数据
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第6张图片
网址输入到浏览器即可
pytorch:使用tensorboardX可视化网络模型时add_graph位置报错_第7张图片

你可能感兴趣的:(pytorch,pytorch,人工智能,python)