span class="p">,</span> <span class="n">section_scores</span>
</span><span id="L-56"><a href="#L-56"><span class="linenos">56</span></a>        <span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">empty_cache</span><span class="p">()</span>
</span><span id="L-57"><a href="#L-57"><span class="linenos">57</span></a>        <span class="n">tq</span><span class="o">.</span><span class="n">update</span><span class="p">(</span><span class="mi">1</s