pan class="n">pretrained</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span>
<span class="n">sparse_hooks</span> <span class="o">=</span> <span class="n">sparsify_model</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="mf">0.9</span><span class="p">,</span> <span class="n">erdos_renyi_sparsity</span><span class="p">)</span>
<span class="n">cbs</span> <span class="o">=</span> <spa