tion </span>
<span class="n">train_config</span> <span class="o">=</span> <span class="n">TrainingConfig</span><span class="p">()</span>
<span class="c1"># Distillation configuration</span>
<span class="c1"># Matching different layers of the student and the teacher</span>
<span class="n">distill_config</span> <span class="o">=</span> <span class="n">DistillationConfig</span><span class="p">(</span>
    <span class="n">intermediate_matches</span><span class="o">=</span><span class="p">[</span>    
     <span