我一直在尝试使用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
添加回答
举报
0/150
提交
取消