Description
看了PaddleServing的例子,按例子作,很简单。但是部署我自己训练的模型,却不知该如何下手。主要的问题是,我不知道模型的输入参数是什么,客户端调用的时候,该如何给参数?
我是基于PaddleHub作的ERNIE的finetue,作文本分类和序列标注。从训练脚本上面看,也看不出模型的输入参数是什么。难道这个东西是固定的吗?该如何确定模型的输入参数和形式呢?谢谢!
训练脚本如下,实在是不知道该如何调用:
注意,Reader、Task、context参数中的max_seq_len应该保持一致
reader = hub.reader.SequenceLabelReader(
dataset = dataset,
vocab_path = module.get_vocab_path(),
max_seq_len = max_seq_len)
strategy = hub.AdamWeightDecayStrategy(
weight_decay = 0.01,
learning_rate = 5e-5,
warmup_proportion = 0.1)
config = hub.RunConfig(
use_cuda = use_cuda,
num_epoch = num_epoch,
batch_size = batch_size,
checkpoint_dir = checkpoint_dir,
eval_interval = 50,
strategy = strategy)
task = hub.SequenceLabelTask(
data_reader = reader,
feature = sequence_output,
feed_list = feed_list,
add_crf = True,
max_seq_len = max_seq_len,
num_classes = dataset.num_labels,
config = config)
Activity