差别
这里会显示出您选择的修订版和当前版本之间的差别。
| 两侧同时换到之前的修订记录前一修订版后一修订版 | 前一修订版 | ||
| 人工智能:paddle:paddle常用脚本:paddle统计模型信息 [2026/01/09 02:52] – [打印paddle模型的基本信息] ctbots | 人工智能:paddle:paddle常用脚本:paddle统计模型信息 [2026/01/09 08:38] (当前版本) – ctbots | ||
|---|---|---|---|
| 行 1: | 行 1: | ||
| ====== paddle统计模型信息 ====== | ====== paddle统计模型信息 ====== | ||
| - | ===== 打印paddle模型的基本信息 ===== | + | ===== 打印paddle的pdparams的基本信息 ===== |
| <code python paddle.py> | <code python paddle.py> | ||
| 行 18: | 行 18: | ||
| # print_paddle_model(" | # print_paddle_model(" | ||
| </ | </ | ||
| + | |||
| + | ===== 打印paddle的模型的基本信息 ===== | ||
| + | |||
| + | 假如我们已经加载了一个paddle的module模型信息,如何打印具体的参数信息。 | ||
| + | |||
| + | <code python p.py> | ||
| + | state_dict = model.state_dict() | ||
| + | for name, tensor in state_dict.items(): | ||
| + | dtype = getattr(tensor, | ||
| + | shape = list(tensor.shape) if hasattr(tensor, | ||
| + | numel = tensor.numel() if hasattr(tensor, | ||
| + | print(f" | ||
| + | </ | ||
| + | |||
| + | |||