Hi @kishwarshafin,
It seems you’ve provided the implementation of your Dataset class to us. Your question however seems to be about the use of DataLoader in an environment with lesser capabilities.
Could you thus please show how you’re using this Dataset implementation with the DataLoader? If possible, share only the relevant code to easen up the reading. 
Petteri