Solved by Antonio_Ossa in post #2 
              
              
                Hi @afshin67, 
If I understand your question corectly, a possible solution is to create a CustomDataset that returns a tuple with label and data or a dictionary containing both, a common approach that is shown in this tutorial. A small example I crafted in case you find it useful: 
class CustomDatas…