Solved by Zhaoyi-Yan in post #4 
              
              
                def my_collfn(batch):
     img = [item[0] for item in batch]
     other_info= [item[1] for item in batch]

     img = torch.stack(img, 0)
     other_info= torch.cat(other_info, dim=0)

     return img, other_info