ride this function if you override train function.
        <li> <a href="#pb_preproc">pre_proc</a> - How image preprocessing is done.
    </ul>

    <h3><a id="pb_create_network"> Defining create_network function:</a></h3>
        Inputs<br>
        <br>
        <!-- <ul>
          <li> None
        </ul> -->
        Outputs
        <ul>
          <li> List of prediction tensors (e.g. heatmaps).
        </ul>
        <p>The tensors containing the training examples are provided as a list in self.inputs.
    