1.https://github.com/sksq96/pytorch-summary
pytorch-summary是一个展示网络结构、参数、输出尺寸的实用分析工具
2.https://github.com/Swall0w/torchstat
torchstat 详细展示网络各层的参数,计算量,内存读写等等
3.https://github.com/Lyken17/pytorch-OpCounter
这是一个简易版的网络分析工具,只显示一个网络的运算量和参数
1.https://github.com/sksq96/pytorch-summary
pytorch-summary是一个展示网络结构、参数、输出尺寸的实用分析工具
2.https://github.com/Swall0w/torchstat
torchstat 详细展示网络各层的参数,计算量,内存读写等等
3.https://github.com/Lyken17/pytorch-OpCounter
这是一个简易版的网络分析工具,只显示一个网络的运算量和参数