-
PyTorch-OpCounter:用于计算PyTorch模型MAC FLOP的工具
资源介绍
思路：PyTorch-OpCounter
如何安装
pip install thop （现在在上持续）
或者
pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git
如何使用
基本用法
from torchvision . models import resnet50
from thop import profile
model = resnet50 ()
input = torch . randn ( 1 , 3 , 224 , 224 )
macs , params = profile ( model , inputs = ( input , ))
定义第三方模块的规则。
class YourModule ( nn . Module ):
# your definition
d