Hi,
I’m not familiar with the USPS dataset, but are the images in there exactly the same size as the mnist dataset?
Also can you use each dataset independently? Do they work properly?
Why do you want to implement in your cistom collate function?