similar error: How does one create a pytorch data loader with a custom hugging face data set without having errors? - #2 by brando