e</span>,</span>
<span>  batch_size <span class="op">=</span> <span class="fl">350</span>,</span>
<span>  num_epochs <span class="op">=</span> <span class="fl">1000</span>,</span>
<span>  learning_rate <span class="op">=</span> <span class="fl">0.001</span></span>
<span><span class="op">)</span></span></code></pre></div>
    </div>

    <div class="section level2">
    <h2 id="arguments">Arguments<a class="anchor" aria-label="anchor" href="#arguments"></a></h2>


<dl><dt id="arg-input-size">input_size<a cla