train_dataset</span> <span class="o">=</span> <span class="n">datasets</span><span class="o">.</span><span class="n">YoloDatasetMultiFile</span><span class="p">(</span>
    <span class="n">train_dataset_path</span><span class="p">,</span> <span class="n">image_shape</span><span class="p">,</span> <span class="n">max_objects</span><span class="p">,</span> <span class="n">batch_shape</span><span class="p">,</span>
    <span class="n">model</span><span class="o">.</span><span class="n">anchors</span><span clas