如何计算 PyTorch 模型中的参数总数?类似于 model.count_params()
在凯拉斯。
最佳答案
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 .如果有人有更好的解决方案,请与我们分享。
关于deep-learning - 检查 PyTorch 模型中的参数总数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/49201236/