="sd">    :type  n_epochs:        int</span>

<span class="sd">    :param optimiser:       The type of optimiser used</span>
<span class="sd">    :type  optimiser:       torch.optim</span>

<span class="sd">    :param batch_size:      The batch size</span>
<span class="sd">    :type  batch_size:       int</span>

<span class="sd">    :param loss_function:   The loss function used</span>
<span class="sd">    :type  loss_function:   torch.nn.modules.loss</span>

<span class="sd">    :param device:          Th