计算FLOPs、Weights、特征可视化

连接参考
https://github.com/TingsongYu/ghostnet_cifar10

bin/03_compute_flops.py

前期准备：安装torchstat

Weights和FLOPs的计算采用torchstat工具，必须是python3以上才可安装

安装方法： pip install torchstat

torchstat网站：https://github.com/Swall0w/torchstat

方法：执行03_compute_flops.py 即可得到Weights和FLOPs

from torchstat import stat
import torchvision.models as models

model = models.resnet18()
stat(model, (3, 224, 224))

特征可视化
脚本：bin/04_fmap_vis.py

方法：运行脚本会在 results/runs下以时间戳为子文件夹记录下events file，然后借助tensorboard就可以查看特征图