the ``module`` is in.</span>

<span class="sd">    Arguments:</span>
<span class="sd">        module (torch.nn.Module): PyTorch module</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <span class="n">params</span> <span class="o">=</span> <span class="n">module</span><span class="o">.</span><span class="n">parameters</span><span class="p">()</span>
    <span class="k">try</span><span class="p">:</span>
        <span class="n">sample_param</span> <span class="o">=</span> <span class="nb">next</span>