模型参数数量的统计

Lemon.灬

已于 2024-01-13 11:20:41 修改

阅读量406

点赞数 11

分类专栏：深度学习以及遇到的各种问题文章标签：人工智能深度学习

于 2024-01-04 16:10:20 首次发布

本文链接：https://blog.csdn.net/weixin_46573159/article/details/135389563

版权

深度学习以及遇到的各种问题专栏收录该内容

7 篇文章 0 订阅

订阅专栏

针对于模型参数数量的统计（.pth文件）

import torch
# Load the model
content = torch.load('xxxx.pth')
model_state_dict = content['model']
# Initialize a variable to count parameters
total_params = 0
# Iterate over the parameters in the model
for param in model_state_dict.values():
    total_params += param.numel()
# Print the total number of parameters
print(f'Total number of parameters in the model: {total_params}')