><a id="pb_train">Defining get_pred_fn</a></h3>
        Inputs<br>
        <ul>
            <li>model_file: Model_file to use. If not specified the latest trained should be used.
        </ul>
        Outputs:<br>
        <ul>
            <li> pred_fn: Create a prediction function that takes a batch of images of size [B,H,W,C] and returns the pose prediction as a python array of size [B, N,2]. The pose prediction must be returned as dictionary with key 'locs'. The input images are raw and need to be preproc