an> <span class="n">eps</span><span class="o">=</span><span class="mf">0.02</span><span class="p">)</span> <span class="c1"># better than 98% accuracy</span>

<span class="k">for</span> <span class="n">m</span> <span class="ow">in</span> <span class="n">sparseable_modules</span><span class="p">(</span><span class="n">learn</span><span class="o">.</span><span class="n">model</span><span class="p">):</span>
    <span class="k">for</span> <span class="n">p</span><span class="p">,</span> <span class="n">mask</s