Pytorch 模型的网络结构可视化

Keras 中 keras.summary() 即可很好的将模型结构可视化,但 Pytorch 暂还没有提供网络模型可视化的工具. 总结两种pytorch网络结构的可视化方法 Pytorch使用Tensorboard可视化网络结构 GitHub地址:点击打开 1.下载可视化代码 git clone https://github.com/lanpa/tensorboard-pytorch.git 2
相关文章
相关标签/搜索