文件名称：pytorch-OpCounter:计算您的PyTorch模型的MAC FLOP
文件大小：14KB
文件格式：ZIP
更新时间：2024-05-12 12:12:52
Python
思路：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
【文件预览】：
pytorch-OpCounter-master
----build.sh(106B)
----.github()
--------workflows()
----requirements.txt(0B)
----benchmark()
--------evaluate_famous_models.py(801B)
--------evaluate_rnn_models.py(2KB)
--------README.md(1KB)
----TODO.md(185B)
----LICENSE(1KB)
----setup.py(1KB)
----README.md(2KB)
----thop()
--------profile.py(7KB)
--------utils.py(681B)
--------__init__.py(121B)
--------vision()
--------rnn_hooks.py(6KB)
----.gitignore(1KB)