type
Post
status
Published
date
Aug 22, 2023
slug
model-params-flops-summary
summary
整理了 PyTorch 模型参数量、可训练参数量和 FLOPs 的常见计算方法，涵盖单入口模型和含编码器/解码器的多入口模型。
tags
开发
知识
category
技术分享
icon
password
本文同步自 CSDN 文章。
第一类：模型一个入口一个出口。
第二类：一个模型多个入口
例如 model下面有两个方法 encoder decoder 这时用上面的方法无法分别进行计算。
使用下面的方法进行计算。
- 计算参数量一般可用：
- 计算GFLOPS可以使用下面的方法：
- 如果想要分别计算不同编码器解码器的参数量可以使用下面的方法。
- 作者:Springli
- 链接:https://blog.5280717.xyz/article/model-params-flops-summary
- 声明:本文采用 CC BY-NC-SA 4.0 许可协议，转载请注明出处。
相关文章
