为了账号安全,请及时绑定邮箱和手机立即绑定

使用 Pegasus 实施迁移学习进行文本摘要生成垃圾字符

使用 Pegasus 实施迁移学习进行文本摘要生成垃圾字符

qq_笑_17 2023-09-12 10:30:14
我一直在尝试使用Pegasus 库生成摘要并按照上述步骤操作 -创建输入数据 .tfrecord于pegasus\data\testdatatransformer_params创建一个返回命名的函数test_transformers(假设)跑步 python3 pegasus/bin/train.py  --params=test_transformer  --param_overrides=vocab_filename=ckpt/pegasus_ckpt/c4.unigram.newline.10pct.96000.model,batch_size=1,beam_size=5,beam_alpha=0.6 --model_dir=ckpt/pegasus_ckpt/xsum/model.ckpt-30000python3 pegasus/bin/evaluate.py  --params=test_transformer  --param_overrides=vocab_filename=ckpt/pegasus_ckpt/c4.unigram.newline.10pct.96000.model,batch_size=1,beam_size=5,beam_alpha=0.6  --model_dir=ckpt/pegasus_ckpt/xsum/model.ckpt-30000但是,当我生成文本时,我在输出中遇到这个问题 -它的实现方式或我在步骤 3 和 4 中运行 python 代码的方式是否存在问题?
查看完整描述

1 回答

?
牧羊人nacy

TA贡献1862条经验 获得超7个赞

此问题突出的原因是:-

1. --model_dir is typically a directory instead of a particular checkpoint. 

   -> Try changing model_dir to actual model directory instead of checkpoint

2. It seems there are only 100 training steps. 

   -> Try changing "train_steps": 100


查看完整回答
反对 回复 2023-09-12
  • 1 回答
  • 0 关注
  • 61 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信