paddlepaddle--paddlenlp/slm/examples/benchmark/wiki_lambada

数据准备

wget https://paddlenlp.bj.bcebos.com/data/benchmark/lambada_test.jsonl

wget https://paddlenlp.bj.bcebos.com/data/benchmark/wikitext-103.tar.gz

wget https://paddlenlp.bj.bcebos.com/data/benchmark/wikitext-2.tar.gz

运行预测脚本

验证 Lambada 数据集，运行以下脚本：

python eval.py \
--model_name_or_path /path/to/your/model \
--batch_size 4 \
--eval_path /path/to/your/dataset/lambada_test.jsonl \
--tensor_parallel_degree 1 \
--cloze_eval

验证 WikiText 数据集，运行以下脚本：

python eval.py \
--model_name_or_path /path/to/your/model \
--batch_size 4 \
--eval_path /path/to/your/dataset/wikitext-103/wiki.valid.tokens \
--tensor_parallel_degree 1