Solved by Jordan_Howell in post #3 
              
              
                Thank you @Nikronic.  I ended up rebuilding it like the below and it seems to work: 
class Inspection_Dataset(Dataset):
        """
        df: Dataframe containing all categorical, numerical and image columns
        numerical columns: list of numerical columns
        cat_columns: list of categori…