https://111qqz.com/en/tags/%E6%A8%A1%E5%9E%8B%E8%BD%AC%E6%8D%A2/Recent content in 模型转换 on 111qqz's blogHugo -- gohugo.ioen© 2015-2026 111qqzFri, 18 Sep 2020 11:02:50 +0800https://111qqz.com/en/post/%E6%B7%B1%E5%BA%A6%E5%AD%A6%E4%B9%A0/2020-09-18-torch2trt-notes/Fri, 18 Sep 2020 11:02:50 +0800https://111qqz.com/en/post/%E6%B7%B1%E5%BA%A6%E5%AD%A6%E4%B9%A0/2020-09-18-torch2trt-notes/<h2 class="relative group">前言 <div id="前言" class="anchor"></div> <span class="absolute top-0 w-6 transition-opacity opacity-0 -start-6 not-prose group-hover:opacity-100 select-none"> <a class="text-primary-300 dark:text-neutral-700 !no-underline" href="#%e5%89%8d%e8%a8%80" aria-label="Anchor">#</a> </span> </h2> <p>偶然发现了 <a href="https://github.com/NVIDIA-AI-IOT/torch2trt" target="_blank" rel="noreferrer">torch2trt</a> 的模型转换方案，思路是直接将pytorch op映射到TensorRT的python api. 在pytorch进行每个op　forward的时候，tensorrt也相应往network上添加op. 这里会先涉及torch2trt的使用，后面会补充这个转换工具的代码学习</p>