开发者社区

文档建议反馈控制台

文章/答案/技术大牛

发布

社区首页 >问答首页 >ONNX和TensorRT模型中的参数和FLOPS的数量

问ONNX和TensorRT模型中的参数和FLOPS的数量
EN

Stack Overflow用户

提问于 2021-11-24 14:23:41

回答 1查看 99关注 0票数 0

将模型从PyTorch转换为ONNX或TensorRT格式时，参数和FLOPS (每秒浮点操作数)的数量是否会发生变化？

EN

回答 1

Stack Overflow用户

发布于 2021-11-24 20:58:25

我知道自从Pytorch的一些新版本(我使用1.8，它对我有效)以来，在保存模型的同时，有一些批量规范层和卷积的融合。我不确定ONNX，但TensorRT积极使用不同层的水平和垂直融合，所以最终的模型会比你初始化的模型计算成本更低。

票数 0

EN

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

原文链接：

https://stackoverflow.com/questions/70097798

复制

相似问题