jdhao:

collate_fn=my_collate


How can I use this method when my original and target are images, like pix2pix or semantic segmentation tasks? I asked this 78 question someone referred to this post for a detailed explanation.