读取pth文件获取参数数量

当你想要从.pth（通常用于PyTorch模型保存）文件中读取参数数量时，首先需要加载这个文件到内存中，这通常涉及到使用`torch.load()`函数。这个函数会返回一个字典，其中包含了模型的参数状态。

例如：

model_state_dict = torch.load('path_to_your_model.pth')

接下来，你可以通过检查`model_state_dict`的键（键通常是参数的名字）数量来获取参数的数量。因为每个键代表一个参数，所以你可以这样做：

num_params = len(model_state_dict)
print(f"模型参数总数: {num_params}")

如果你想要得到特定类型的参数数量，比如所有权重（weights）或者偏置（bias），可以遍历字典并计数相应类型的关键字：

weight_keys = [key for key in model_state_dict if 'weight' in key]
bias_keys = [key for key in model_state_dict if 'bias' in key]

num_weights = len(weight_keys)
num_biases = len(bias_keys)

print(f"权重参数数量: {num_weights}, 偏置参数数量: {num_biases}")