Pytorch的网络结构可视化(tensorboardX)(详细)

最近刚刚发现一个很是好用的显示模型神器Netronhtml https://github.com/lutzroeder/Netrongit https://www.electronjs.org/apps/netrongithub 借助这个工具能够像windows的软件同样导入已经训练好的模型加权重便可一键生成chrome 如下是个人一个模型使用该工具可视化结果,只不过目前该工具对于onnx支持很是
相关文章
相关标签/搜索