ser:           torch.optim</span>
<span class="sd">        </span>
<span class="sd">    :param num_epoch:           The number of epochs run for</span>
<span class="sd">    :type  num_epoch:           int</span>
<span class="sd">    </span>
<span class="sd">    :param learning_rate:       The learning rate learning hyper-parameter</span>
<span class="sd">    :type  learning_rate:       float</span>
<span class="sd">    </span>
<span class="sd">    :param momentum:            The momentum learning hyper-para