pan> <span class="o">==</span> <span class="s1">&#39;Image&#39;</span><span class="p">:</span>
            <span class="n">dataset</span> <span class="o">=</span> <span class="n">ImageDataset</span><span class="p">(</span>
                <span class="n">config</span><span class="o">.</span><span class="n">path</span><span class="p">[</span><span class="n">i</span><span class="p">],</span> <span class="n">config</span><span class="o">.</span><span class="n">split</span><span class="p">[</span><span class="n