By the way, here is a reference to the issue I had. I was thinking that it is somehow related to data augmentation, i.e., the landmarks are falling outside of the image after various image transformations. However, then I’ve got another error:
UserWarning:
There seems to be something wrong with your dataset, can't access self.train_ds[i] for all i in
[65237, 47545, 8078, 53990, ..., 758]
I am going to try to reproduce this issue on some small/dummy dataset to see if it still exists in the library.