PaddleNLP_roberta/tests/benchmark

• PaddleNLP Benchmark 测试脚本

PaddleNLP Benchmark 测试脚本

目前我们为用户提供了GPT模型的Benchmark性能测试脚本。

启动测试脚本的方法如下：

ImageName="registry.baidubce.com/paddlepaddle/paddle:2.1.2-gpu-cuda10.2-cudnn7"
docker pull ${ImageName}

run_cmd="set -xe;
        cd /workspace ;
        bash -x tests/benchmark/run_all.sh"

nvidia-docker run --name test_paddle_gpt -i  \
    --net=host \
    --shm-size=1g \
    -v $PWD:/workspace \
    ${ImageName}  /bin/bash -c "${run_cmd}"

如果需要打开profile选项，可以直接替换run_cmd为：

run_cmd="set -xe;
        cd /workspace ;
        bash -x tests/benchmark/run_all.sh on"