Pytorch 第三方工具:模型输出、可视化
1. 模型信息输出:torchinfo
库
torchinfo
库可是输出模型的结构、参数等信息:GitHub
实例:
from torchinfo import summary
model = ConvNet()
batch_size = 16
summary(model, input_size=(batch_size, 1, 28, 28))
2. 网络结构可视化
ML 模型可视化模板:GitHub;下载地址:Google Powerpoint
tensorwatch
:GitHub
torchviz
:GitHub