Hi, based on your code snippet, it doesn’t seem like your __getitem__ method is returning anything. Maybe try adding return torch_data at the end?
If you include your error message here, it will be helpful as well.