e device to run on (Cpu or CUDA)</span>
<span class="sd">    :type  device:          string</span>

<span class="sd">    :param seed:            The number that is set for the random seeds</span>
<span class="sd">    :type  seed:            int</span>

<span class="sd">    :param debug:           Whether to print some parameters for checking</span>
<span class="sd">    :type debug:            bool</span>

<span class="sd">    :param disp_freq:       The epoch frequency that training/validation</span>
<span 