这是本文档旧的修订版!
1-paddle模型相关的api和操作
打印paddle模型的基本结构信息
import paddle
state = paddle.load("/home/ctbots/llm/paddle/7b/model_state.pdparams")
for k, v in state.items():
print(
f"{k:60s} | "
f"dtype={v.dtype} | "
f"shape={list(v.shape)} | "
f"numel={v.numel()}"
)
评论