nn">torch</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">torch_tensorrt</span>
<span class="kn">import</span><span class="w"> </span><span class="nn">torchvision.models</span><span class="w"> </span><span class="k">as</span><span class="w"> </span><span class="nn">models</span>
</pre></div>
</div>
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="c1"># Initialize model with half precision and sample inputs</span>
<span class="n">mo