检查PyTorch模型中的参数总数

作者: 狗头军师
发布时间: 2025-02-20 02:04:07 (18天前)

2 条回复

1#
回复此人
v-star*위위 | 2020-08-21 17-46

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) 受PyTorch论坛上此答案启发的答案。

编辑

登录后才能参与评论