logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

Pytorch的网络可视化

概述有时候想要绘制一个神经网络的网络结构,可以通过代码中层的定义去用第三方软件依次绘制,但是这样费时费力。因此需要一种快速绘制的办法。Netronnetron可通过所保存的模型将其用网络的方式可视化出来,但是对于pytorch来说,其支持程度还不够,无法绘制各参数间的关系,可以将pytorch模型导出为onnx格式再使用netron来可视化。# 自建一个CNN模型并导出为onnx来可视化impor

#python#pytorch#神经网络
到底了