请问如何保存并部署一个hub.Module加载出来的模型
收藏
我通过fine-tune已经训练好了一个paddlehub里面的模型,但是应该怎么保存它然后部署呢?
model = hub.Module( name='ernie_tiny', version='2.0.1', task='seq-cls', load_checkpoint='/home/temp1/NLP/test_ernie_text_cls/best_model/model.pdparams', label_map=label_map)
0
收藏
请登录后评论