Skip to content

add resnet50 trt tests in pr-ci-inference#34465

Merged
OliverLPH merged 18 commits intoPaddlePaddle:developfrom
OliverLPH:enable_infer_tests
Jul 31, 2021
Merged

add resnet50 trt tests in pr-ci-inference#34465
OliverLPH merged 18 commits intoPaddlePaddle:developfrom
OliverLPH:enable_infer_tests

Conversation

@OliverLPH
Copy link
Contributor

@OliverLPH OliverLPH commented Jul 28, 2021

PR types

Others

PR changes

Others

Describe

  1. PR-CI-INFERENCE 流水线中开启TensorRT编译
  2. 增加了 infer_ut 目录和gtest依赖。
  3. 增加了 ResNet50 模型的3种测试,打开IR的GPU前向推理TRT FP32预测,batch_size=2预测TRT FP32预测,序列输入batch_size=1,2,3,4的数据预测。与关闭IR的GPU前向推理进行结果校验。该test仅在PR-CI-INFERENCE流水线中生效。
  4. demo_ci 中的 trt_mobilenet_demo 在打开 trt 预测后会失败,待修复

@paddle-bot-old
Copy link

paddle-bot-old bot commented Jul 28, 2021

✅ This PR's description meets the template requirements!
Please wait for other CI results.

@paddle-bot-old
Copy link

Thanks for your contribution!
Please wait for the result of CI firstly. See Paddle CI Manual for details.

jiweibo
jiweibo previously approved these changes Jul 29, 2021
Copy link
Contributor

@jiweibo jiweibo left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM,期待后续把windows的运行脚本也加下哈

@OliverLPH OliverLPH changed the title add resnet50 trt tests in demo_ci add resnet50 trt tests in pr-ci-inference Jul 29, 2021
Copy link
Contributor

@jiweibo jiweibo left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me.

@OliverLPH OliverLPH merged commit 8b72a1a into PaddlePaddle:develop Jul 31, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants