>    net: torch.nn.Module</span>
<span class="sd">        The deep model to use</span>
<span class="sd">    nclasses: int</span>
<span class="sd">        Number of unique values for the target</span>
<span class="sd">    args: dict</span>
<span class="sd">        Specify additional parameters</span>
<span class="sd">        </span>
<span class="sd">        - **batch_size**: Batch size to be used inside strategy class (int, optional)</span>
<span class="sd">        - **device**: The device that this 