esmm模型保存模型而不是模型参数遇到问题

PaddlePaddle / PaddleRec

Recommendation Algorithm大规模推荐算法库，包含推荐系统经典及最新算法LR、Wide&Deep、DSSM、TDM、MIND、Word2Vec、Bert4Rec、DeepWalk、SSR、AITM，DSIN，SIGN，IPREC、GRU4Rec、Youtube_dnn、NCF、GNN、FM、FFM、DeepFM、DCN、DIN、DIEN、DLRM、MMOE、PLE、ESMM、ESCMM, MAML、xDeepFM、DeepFEFM、NFM、AFM、RALM、DMR、GateNet、NAML、DIFM、Deep Crossing、PNN、BST、AutoInt、FGCNN、FLEN、Fibinet、ListWise、DeepRec、ENSFM，TiSAS，AutoFIS等，包含经典推荐系统数据集criteo 、movielens等

https://paddlerec.readthedocs.io/

Apache License 2.0

4.19k stars 723 forks source link

esmm模型保存模型而不是模型参数遇到问题 #723

Open FelixLiu1996 opened 2 years ago

FelixLiu1996 commented 2 years ago

部署服务需要用到模型而不是模型参数，所以想要保存模型使用paddle.save(net, model_prefix + ".pdmodel") 显示报错 ValueError: paddle do not support saving paddle.nn.Layer object.

使用 paddle.jit.save(net, model_prefix + ".pdmodel",input_spec=None) 显示报错 ValueError: No valid transformed program for function: forward(inputs), input_spec: None. Please specific input_spec in @paddle.jit.to_static or feed input tensor to call the decorated function at once.

请问这个该怎么解决呢

yinhaofeng commented 2 years ago

想要部署的话，可以直接使用save_inference_model接口保存，推荐试一下这个教程：https://github.com/PaddlePaddle/PaddleRec/blob/master/doc/serving.md