deep-learning - 检查 PyTorch 模型中的参数总数

Question

如何计算 PyTorch 模型中的参数总数？类似于 model.count_params()在凯拉斯。

Answer 1

PyTorch 没有像 Keras 那样计算参数总数的函数，但可以对每个参数组的元素数求和:

pytorch_total_params = sum(p.numel() for p in model.parameters())

如果只想计算可训练参数:

pytorch_total_params = sum(p.numel() for p in model.parameters() if p.requires_grad)

受此启发的答案 answer在 PyTorch 论坛上。

注意:我是 answering my own question .如果有人有更好的解决方案，请与我们分享。