an class="o">.</span><span class="n">output_names</span><span class="p">,</span>
                <span class="n">target_device</span><span class="o">=</span><span class="n">Device</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;cuda:</span><span class="si">{</span><span class="n">torch</span><span class="o">.</span><span class="n">cuda</span><span class="o">.</span><span class="n">current_device</span><span class="p">()</span><span class="si">}</span><span class="s2">&quot;</sp