这是本文档旧的修订版！
paddle统计模型信息
打印paddle模型的基本信息
import paddle
def print_paddle_model(model_path):
state = paddle.load(model_path)
for k, v in state.items():
print(
f"{k:60s} | "
f"dtype={v.dtype} | "
f"shape={list(v.shape)} | "
f"numel={v.numel()}"
)
# print_paddle_model("xxx/model_state.pdparams")
评论