an class="c1"># TODO need to generalize for n-class classifier</span>
        <span class="k">return</span> <span class="n">out</span></div>


<div class="viewcode-block" id="PyTorchModel.set_eval_mode">
<a class="viewcode-back" href="../../../dice_ml.model_interfaces.html#dice_ml.model_interfaces.pytorch_model.PyTorchModel.set_eval_mode">[docs]</a>
    <span class="k">def</span><span class="w"> </span><span class="nf">set_eval_mode</span><span class="p">(</span><span class="bp">self</span><span class="p">)