Hi, sorry. That’s the mistake that I made. You are right. Each batch is a tuple that each element of this tuple is a tensor of shape [batch_size, h,w,d]. Wrongly, I thought the length of batch should be the same as the batch_size that we pass to the data loader. Thanks for the point.