静态量化时,模型格式问题
收藏
我用的是RK1808的板子,在学习PaddleSlim-quant-demo.tar.gz 这个demo 时,model里的模型跟paddledection 里训练好的模型(如pdmodel 和pdiparams) 格式似乎不一样,去运行自己训练好的模型作量化时就报错了,运行demo里的mobilenet_v1 和 resnet50时是正确的,用自己的训练 好的模型做静态量化时还需要注意什么? 谢谢
0
收藏
请登录后评论
你好,请问问题解决了吗?没解决的话,建议在Paddle Slim的github上提个issue,或者你加我微信,我拉你进Paddle slim群。
paddle 一共包含 *.pdmodel,*.pdparams,*.pdop,*.pdiparams四个文件。*.pdmodel是训练使用的完整模型program描述,区别于推理模型,训练模型program包含完整的网络,包括前向网络,反向网络和优化器,而推理模型program仅包含前向网络,*.pdparams是训练网络的参数dict,key为变量名,value为Tensor array数值,*.pdopt是训练优化器的参数,结构与*.pdparams一致,*.pdiparams为模型推理用的参数,二进制Tensor存储格式,不含变量名。以上使用不同的api生成,你可以根据需要的格式在官网文档查看。
解决了,谢谢