&#39;adam&#39;</span><span class="p">,</span> <span class="mf">1e-4</span><span class="p">)</span>
<span class="n">model_loss</span> <span class="o">=</span> <span class="n">model</span><span class="o">.</span><span class="n">get_loss_function</span><span class="p">(</span><span class="n">num_batches</span> <span class="o">=</span> <span class="nb">len</span><span class="p">(</span><span class="n">train_dataset</span><span class="p">))</span>
</pre></div>
</div>
<div class="highlight-default notranslate"><d