ate"><div class="highlight"><pre><span></span><span class="n">Train</span> <span class="k">for</span> <span class="mi">2</span> <span class="n">steps</span><span class="p">,</span> <span class="n">validate</span> <span class="k">for</span> <span class="mi">2</span> <span class="n">steps</span>
<span class="n">Epoch</span> <span class="mi">1</span><span class="o">/</span><span class="mi">5</span>
<span class="mi">2</span><span class="o">/</span><span class="mi">2</span> <span class="p">[</span><span class="o