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

3. Early Stop 技巧