C++&TensorRT | 推理时动态shape

动态shape,即编译时指定可动态的范围[L-H],推理时可以允许 L<= shape <= H
OptimizationProfile是一个优化配置文件,用来指定输入的shape可以变换的范围的,不要被优化两个字
蒙蔽了双眼

  1. 构建网络时:
    • 1.1. 必须在模型定义时,输入维度给定为-1,否则该维度不会动态。注意一下两点:
      • 1.1.1. 若onnx文件,则onnx文件打开后应该看到为动态或者-1
      • 1.1.2. 如果你的模型中存在reshape类操作,那么reshape的参数必须随动态进行计算。而大部分时候这都是问题。除非你是全卷积模型,否则大部分时候只需要为batch_size维度设置为动态,其他维度尽量避免设置动态
    • 1.2. 配置profile:
      • 1.2.1. create: builder->createOptimizationProfile()
      • 1.2.2. set: setDimensions()设置kMIN, kOPT, kMAX的一系列输入尺寸范围
      • 1.2.3. add:config->addOptimizationProfile(profile);添加profile到网络配置中
  2. 推理阶段时:
    • 2.1. 您需要在选择profile的索引后设置input维度:execution_context->setBindingDimensions(0, nvinfer1::Dims4(1, 1, 3, 3));
        <
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

ASKCOS

你的鼓励是我最大的动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值