e at each landmarks predicted location.
        <li> <a href="#pb_loss">loss</a> - Define the loss function to be optimized.
    </ul>
    <p>In addition, if you want to change the training procedure, you'll have to override</p>
    <ul>
        <li> <a href="#pb_train">train</a> -If you want to different learning rate schedule or optimizer.
        <li> <a href="#pb_get_pred_fn">get_pred_fn</a> - To create a prediction function that will take a batch of input images and return the predictions on them. Over