模型可视化工具和库

小结模型可视化工具和库

Thop

仓库链接:Lyken17/pytorch-OpCounter

Thop能够计算模型参数数目以及Flops。其具体实现参考参数数目和Flops计算

Netscope CNN Analyzer

Netscope CNN Analyzer是一个在线可视化网络模型网站,支持Caffeprototxt格式,也提供了很多模型的实现

pytorchviz

仓库链接:szagoruyko/pytorchviz

pytorchviz能够生成PyTorch模型的执行图

torchsummary

仓库链接:sksq96/pytorch-summary

torchsummary能够输出模型每一层的参数数目

相关阅读