: torch.Size([128, 3, 224, 224])
Output features size: torch.Size([128, 1000])
Average batch time: 41.21 ms
</pre></div></div>
</div>
</section>
<section id="FP16-(half-precision)">
<h3>FP16 (half precision)<a class="headerlink" href="#FP16-(half-precision)" title="Permalink to this headline">¶</a></h3>
<div class="nbinput docutils container">
<div class="prompt highlight-none notranslate"><div class="highlight"><pre><span></span>[11]:
</pre></div>
</div>
<div class="input_area highlight-ipython3 notransla