温馨提示×

pytorch怎么查看参数数量

小亿
441
2024-01-09 17:02:21
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

使用pytorch库中的model.parameters()可以获得模型的所有参数,然后使用len()函数可以统计参数的数量。下面是一个示例代码:

import torch
import torch.nn as nn

# 创建模型
model = nn.Linear(10, 5)

# 统计参数数量
num_parameters = sum(p.numel() for p in model.parameters())
print(f"模型参数数量: {num_parameters}")

输出结果会显示模型的参数数量。

亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读:怎么查看tensorflow的参数量

0