首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >要让多个批处理在TensorFlow Lite中正常工作,需要满足哪些条件?

要让多个批处理在TensorFlow Lite中正常工作,需要满足哪些条件?
EN

Stack Overflow用户
提问于 2021-06-27 14:40:15
回答 1查看 46关注 0票数 1

您知道,在某些情况下,通过在推理之前调用ResizeInputTensor,TensorFlow Lite可以很好地处理多批推理。但是我的模型不能很好地工作。

要让多个批处理在TensorFlow Lite中正常工作,需要满足哪些条件?

我怀疑您需要将batch维度设置为-1才能进行整形。

我想知道确切的情况。

谢谢。

EN

回答 1

Stack Overflow用户

发布于 2021-06-30 13:04:35

为了能够具有动态批处理,原始模型必须具有动态批处理,然后当您转换为tflite时,导出的模型将支持调整批处理大小。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/68148631

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档