会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
博客园
首页
新随笔
联系
订阅
管理
2024年12月18日
模型参数量计算(以transformer为例)
摘要: 前言 模型中常见的可训练层包括卷积层和线性层,这里将给出计算公式并在pytorch下进行验证。 计算模型的参数: import torch.nn as nn def cal_params(model: nn.Module): num_learnable_params = sum(p.numel()
阅读全文
posted @ 2024-12-18 17:10 片刻的自由
阅读(1908)
评论(0)
推荐(0)