># Enabled precision for TensorRT optimization</span>
<span class="n">enabled_precisions</span> <span class="o">=</span> <span class="p">{</span><span class="n">torch</span><span class="o">.</span><span class="n">half</span><span class="p">}</span>


<span class="c1"># Workspace size for TensorRT</span>
<span class="n">workspace_size</span> <span class="o">=</span> <span class="mi">20</span> <span class="o">&lt;&lt;</span> <span class="mi">30</span>

<span class="c1"># Maximum number of TRT Engines</span>
<