1. torch .pt -> netron
netron是一个专门可视化模型的工具,支持很多格式,很自然的就是用它直接显示torch保存的模型。但是实际上...
2. 网上说的比较多的几种方式
a. tensorboardx 画出来的比较丑 b. tensorwatch 支持的torch版本低,容易出错 c. graphviz 实测yolov5异常。。3. torch->caffe->netron
caffe支持的操作不够多,转换很容易碰到不支持的情况4. torch->onnx->netron
pytorch对onnx支持的比较好,直接有导出接口。 torch.onnx.export... 但这样直接转出来的onnx进行显示没有每一层的输入大小,如下图,还是不方便看。