 class="n">student_model</span><span class="p">,</span><span class="n">max_level</span><span class="o">=</span><span class="mi">3</span><span class="p">)</span>

<span class="c1"># Define an adaptor for translating the model inputs and outputs</span>
<span class="k">def</span> <span class="nf">simple_adaptor</span><span class="p">(</span><span class="n">batch</span><span class="p">,</span> <span class="n">model_outputs</span><span class="p">):</span>
  	<span class="c1"># The second and third elements of mo