关于模型参数量

[TingFinn]

您好，请问您是使用什么方法计算模型参数量的呢?

[Bobholamovic]

可以在构建模型后使用如下方式统计可训练参数量：

print(sum(1 for p in model.parameters() if p.requires_grad))

也可以使用第三方库统计参数量和计算量，如flops-counter.pytorch和THOP。