    kwargs : dict</span>
<span class="sd">        Extra parameters for the model</span>

<span class="sd">    Returns</span>
<span class="sd">    -------</span>
<span class="sd">    model : nn.Module</span>
<span class="sd">        Created model</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <span class="n">print0</span><span class="p">(</span><span class="n">pcolor</span><span class="p">(</span><span class="s1">&#39;Model: </span><span class="si">%s</span><span class="s1">&#39;</span> <span clas