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