search

PaddlePaddle / PaddleHub

Awesome pre-trained models toolkit based on PaddlePaddle. (400+ models including Image, Text, Audio, Video and Cross-Modal with Easy Inference & Serving)【安全加固，暂停交互，请耐心等待】
https://www.paddlepaddle.org.cn/hub
Apache License 2.0
12.75k stars 2.07k forks source link

[Cherry-pick] Fix save_inference_model bug in paddlehub (#2006) #2143

Closed rainyfly closed 2 years ago

rainyfly commented 2 years ago

修复paddlehub module.py中的save_inference_model对于静态图模型完全没法使用的问题。 目前主要存在如下两个问题：

  1. moduleinfo 这个decorator通过type动态创建新类，基类使用hub.RunModule, 属性直接使用用户自定义module类的dict，没有dict中的'dict' 去掉，导致访问instance的dict属性时会直接报错。
  2. save_inference_model函数中调用paddle.static.save_inference_model, 参数名称完全不对，应该是从fluid接口迁移时候只改了接口名，没改接口参数。