请问如何保存并部署一个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
收藏
请登录后评论