pytoch可视化网络工具-pytorchviz

tensorflow的可视化工具是tensorboard,但是tensorboard在显示网络结构图的时候,显示特别混乱. 之前看了caffe的可视化网络图,很清晰,最近发现pytorch的网络图可视化工具,对网络图的显示也很清晰. 第一步:安装: 一行命令安装 conda install -c anaconda graphviz python-graphviz 第二步:可视化代码(最好自己单独写
相关文章
相关标签/搜索