an><span class="bp">self</span><span class="o">.</span><span class="n">test_dataset</span><span class="p">,</span>
                                <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">datasets</span><span class="o">.</span><span class="n">test</span><span class="p">,</span> <span class="s1">&#39;test&#39;</span><span class="p">)</span></div>

<div class="viewcode-block" id="ModelWrapper.training_step"><a class="viewcode-back