><span class="kn">import</span> <span class="n">BaseModel</span>


<div class="viewcode-block" id="PyTorchModel">
<a class="viewcode-back" href="../../../dice_ml.model_interfaces.html#dice_ml.model_interfaces.pytorch_model.PyTorchModel">[docs]</a>
<span class="k">class</span><span class="w"> </span><span class="nc">PyTorchModel</span><span class="p">(</span><span class="n">BaseModel</span><span class="p">):</span>

    <span class="k">def</span><span class="w"> </span><span class="fm">__init__</span><span c