PyTorchでパラメータ数を確認する方法は？

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}")

モデルのパラメータ数が表示されます。