文件名称：flops-counter.pytorch:pytorch 框架中卷积网络的触发器计数器
文件大小：14KB
文件格式：ZIP
更新时间：2024-08-13 11:41:01
deep-neural-networks pytorch deeplearning pytorch-cnn pytorch-utils
pytorch 框架中卷积网络的触发器计数器 该脚本旨在计算卷积神经网络中乘加运算的理论量。 它还可以计算参数数量并打印给定网络的每层计算成本。 支持的层： Conv1d/2d/3d（包括分组） ConvTranspose1d/2d/3d（包括分组） BatchNorm1d/2d/3d、GroupNorm、InstanceNorm1d/2d/3d 激活（ReLU、PReLU、ELU、ReLU6、LeakyReLU） 线性 上采样 池化（AvgPool1d/2d/3d、MaxPool1d/2d/3d 和自适应池） 实验支持： RNN、LSTM、GRU（假设为 NLH 布局） RNNCell、LSTMCell、GRUCell 多头注意力 要求：Pytorch >= 1.1，torchvision >= 0.3 感谢@warmspringwinds 提供脚本的初始版本。
【文件预览】：
flops-counter.pytorch-master
----ptflops()
--------__init__.py(53B)
--------flops_counter.py(18KB)
----.flake8(127B)
----.gitignore(1KB)
----README.md(9KB)
----CHANGELOG.md(1KB)
----LICENSE(1KB)
----.isort.cfg(359B)
----setup.py(754B)
----test_requirements.txt(75B)
----.travis.yml(241B)
----samples()
--------bert.py(1KB)
--------classification.py(1KB)