span>
                        <span class="n">worker_init_fn</span><span class="o">=</span><span class="n">worker_init_fn</span><span class="p">,</span>
                        <span class="n">sampler</span><span class="o">=</span><span class="n">get_datasampler</span><span class="p">(</span><span class="n">dataset</span><span class="p">,</span> <span class="n">mode</span><span class="p">))</span>
             <span class="p">)</span> <span class="k">for</span> <span class="n">dataset</span> <span class="ow