ss="n">accelerator</span><span class="p">,</span>
        <span class="n">strategy</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">strategy</span><span class="p">,</span>
        <span class="n">precision</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span class="n">precision</span><span class="p">,</span>
        <span class="n">devices</span><span class="o">=</span><span class="n">args</span><span class="o">.</span><span c