d">&quot;&quot;&quot;</span>
<span class="sd">    Track model parameters and gradients.</span>

<span class="sd">    Arguments:</span>
<span class="sd">        model (Optimizer): PyTorch model</span>

<span class="sd">    Keyword Arguments:</span>
<span class="sd">        model_name (str, optional): name of the model</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <span class="k">for</span> <span class="n">name</span><span class="p">,</span> <span class="n">param</span> <span class="ow">in</span> 