diff --git a/PaddleNLP/examples/text_generation/couplet/README.md b/PaddleNLP/examples/text_generation/couplet/README.md index 8bc708178d3bc77a8c93e4396f82f666a48e1041..febdd2a5c100673be75aaf2e35a98af9a87c3864 100644 --- a/PaddleNLP/examples/text_generation/couplet/README.md +++ b/PaddleNLP/examples/text_generation/couplet/README.md @@ -28,7 +28,7 @@ Sequence to Sequence (Seq2Seq),使用编码器-解码器(Encoder-Decoder) ## 数据介绍 -本教程使用[couplet数据集](https://bj.bcebos.com/paddlehub-dataset/couplet.tar.gz)数据集作为训练语料,train.tsv作为训练集,dev.tsv数据作为开发集,test.tsv数据作为测试集 +本教程使用[couplet数据集](https://paddlenlp.bj.bcebos.com/datasets/couplet.tar.gz)数据集作为训练语料,train_src.tsv及train_tgt.tsv为训练集,dev_src.tsv及test_tgt.tsv为开发集,test_src.tsv及test_tgt.tsv为测试集。 数据集会在`CoupletDataset`初始化时自动下载 @@ -48,6 +48,7 @@ python train.py \ 各参数的具体说明请参阅 `args.py` 。训练程序会在每个epoch训练结束之后,save一次模型。 +**NOTE:** 如需恢复模型训练,则`init_from_ckpt`只需指定到文件名即可,不需要添加文件尾缀。如`--init_from_ckpt=couplet_models/19`即可,程序会自动加载模型参数`couplet_models/19.pdparams`,也会自动加载优化器状态`couplet_models/19.pdopt`。 ## 模型预测 @@ -62,27 +63,26 @@ python predict.py \ --infer_output_file infer_output.txt \ --beam_size 10 \ --use_gpu True - ``` 各参数的具体说明请参阅 `args.py` ,注意预测时所用模型超参数需和训练时一致。 ## 生成对联样例 -崖悬风雨骤 月落水云寒 +上联:崖悬风雨骤 下联:月落水云寒 -约春章柳下 邀月醉花间 +上联:约春章柳下 下联:邀月醉花间 -箬笠红尘外 扁舟明月中 +上联:箬笠红尘外 下联:扁舟明月中 -书香醉倒窗前月 烛影摇红梦里人 +上联:书香醉倒窗前月 下联:烛影摇红梦里人 -踏雪寻梅求雅趣 临风把酒觅知音 +上联:踏雪寻梅求雅趣 下联:临风把酒觅知音 -未出南阳天下论 先登北斗汉中书 +上联:未出南阳天下论 下联:先登北斗汉中书 -朱联妙语千秋颂 赤胆忠心万代传 +上联:朱联妙语千秋颂 下联:赤胆忠心万代传 -月半举杯圆月下 花间对酒醉花间 +上联:月半举杯圆月下 下联:花间对酒醉花间 -挥笔如剑倚麓山豪气干云揽月去 落笔似龙飞沧海龙吟破浪乘风来 +上联:挥笔如剑倚麓山豪气干云揽月去 下联:落笔似龙飞沧海龙吟破浪乘风来