首页 PaddleHub 帖子详情
paddlehub内置ERNIE-gen生成文本怎么只有50个字,更改哪个超参能输出变长
收藏
快速回复
PaddleHub 问答迁移学习预训练模型 449 2
paddlehub内置ERNIE-gen生成文本怎么只有50个字,更改哪个超参能输出变长
收藏
快速回复
PaddleHub 问答迁移学习预训练模型 449 2

paddlehub内置ERNIE-gen生成文本摘要问什么只有50个字,更改哪个超参能输出变长?

API
参数
train_path(str):
训练集路径。训练集的格式应为:"序号\t输入文本\t标签",例如:"1\t床前明月光\t疑是地上霜"
dev_path(str):
验证集路径。验证集的格式应为:"序号\t输入文本\t标签",例如:"1\t举头望明月\t低头思故乡"
save_dir(str):
模型保存以及验证集预测输出路径。
init_ckpt_path(str):
模型初始化加载路径,可实现增量训练。
use_gpu(bool):
是否使用GPU。
max_steps(int):
最大训练步数。
batch_size(int):
训练时的batch大小。
max_encode_len(int):
最长编码长度。
max_decode_len(int):
最长解码长度。
learning_rate(float):
学习率大小。
warmup_proportion(float):
学习率warmup比例。
weight_decay(float):
权值衰减大小。
noise_prob(float):
噪声概率,详见ernie gen论文。
label_smooth(float):
标签平滑权重。
beam_width(int):
验证集预测时的beam大小。
length_penalty(float):
验证集预测时的长度惩罚权重。
log_interval(int):
训练时的日志打印间隔步数。
save_interval(int):
训练时的模型保存间隔部署。验证集将在模型保存完毕后进行预测。

0
收藏
回复
全部评论(2)
时间顺序
大分
#3 回复于2022-06

模型里面有个decode.py的几个文件,可以修改长度,导出模型的时候也可以指定长度

0
回复
竹影清风夜鸢
#4 回复于2022-12

1

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户