模型可视化工具和库
小结模型可视化工具和库
Thop
仓库链接:Lyken17/pytorch-OpCounter
Thop能够计算模型参数数目以及Flops。其具体实现参考参数数目和Flops计算
Netscope CNN Analyzer
Netscope CNN Analyzer是一个在线可视化网络模型网站,支持Caffe的prototxt格式,也提供了很多模型的实现
pytorchviz
pytorchviz能够生成PyTorch模型的执行图
torchsummary
torchsummary能够输出模型每一层的参数数目