If you still have any problem, feel free to reply and I’ll send you some examples which I have implemented myself.
Here is the question 4 I asked for the first time I was working with custom Dataset class.
I hope it helps you