文件名称：Cal-FLOPs-for-PLM:在NLP中计算预训练模型的FLOP
文件大小：2KB
文件格式：ZIP
更新时间：2024-04-18 13:27:05
natural-language-processing parameters transformers pytorch flops
Cal-FLOPs for PLM 在NLP中计算预训练模型的FLOP 该存储库提供了有关为Pytorch Framework中的NLP模型（主要是PLM）计算FLOP和参数的示例脚本。 该示例脚本展示了两种类型的开源FLOPs计数器工具的用法。 FLOPs计数器和Parameters计数器基于和的开源工具 安装 从PyPI： pip install thop pip install ptflops 要求： 火炬> = 1.0.0 变压器> = 2.2.0 例子 参见example.py
【文件预览】：
Cal-FLOPs-for-PLM-main
----example.py(4KB)
----README.md(680B)