Solved by crcrpar in post #2 
              
              
                Hi, 
I think making Dataset instance to return a batch of metadata with images (and targets) is a way if the meta-features can be represented in torch.Tensor. 
For example, if we use the dataset defined below, in each iteration a DataLoader will generate a tuple of (image, target, metas), where each…