The error message points towards an empty sample in your batch so check if you are properly selecting the individuals in your __getitem__ method.
1 Like
The error message points towards an empty sample in your batch so check if you are properly selecting the individuals in your __getitem__ method.