an class="sd">    :param model_name:     The model name to load</span>
<span class="sd">    :type  model_name:     string</span>

<span class="sd">    :param path:           The directory path to load the model from</span>
<span class="sd">    :type  path:           string</span>
<span class="sd">    &quot;&quot;&quot;</span>
    <span class="n">model</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">path</s