ass="s1">&#39;loss&#39;</span><span class="p">:</span> <span class="s1">&#39;hidden_mse&#39;</span><span class="p">,</span><span class="s1">&#39;weight&#39;</span> <span class="p">:</span> <span class="mi">1</span><span class="p">}])</span>

<span class="c1"># Build distiller</span>
<span class="n">distiller</span> <span class="o">=</span> <span class="n">GeneralDistiller</span><span class="p">(</span>
    <span class="n">train_config</span><span class="o">=</span><span class="n">train_config</span><span cl