 class="kc">None</span><span class="p">:</span>
                    <span class="n">store_l1_l2</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;grad.</span><span class="si">{</span><span class="n">model_name</span><span class="si">}</span><span class="s2">.</span><span class="si">{</span><span class="n">name</span><span class="si">}</span><span class="s2">&quot;</span><span class="p">,</span> <span class="n">param</span><span class="o">.</span><span class="n">grad</span><span c