pan>
<span class="sd">        </span>
<span class="sd">        :param train_loader:     The iterator that feeds in the training data</span>
<span class="sd">        :type  train_loader:     torch.utils.data.dataloader.DataLoader</span>

<span class="sd">        :return:                 The error metric for that epoch</span>
<span class="sd">        :rtype:                  float</span>
<span class="sd">        &quot;&quot;&quot;</span>
        <span class="c1"># Sets the model to train mode</span>
        <