Solved by Nikronic in post #2 
              
              
                Please read the documentation and look for “custom dataset pytorch” in forum. This question has a lot of answers already. 
Data Loading Tutorial