Solved by Alexey_Demyanchuk in post #5 
              
              
                Hey Prashanth, let me try to explain a concept of dataset and dataloader in pytorch as I understand it. 
Dataset is a whole set of your data points (samples) you want to iterate through during training. It defines a logic on how to access the sample at some index idx. Dataloader  - is a helper lets …