I am trying to train a keypoint detection model for my application. I have gone through labelling, but am having issues with the Colab notebook provided for training. The details for the error are posted below, and I am experiencing this with both the default dataset and my custom dataset.