我一直在嘗試使用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
提交
取消