Hi,
I was curious, how do you deal when one data set has 3 channels (imagenet) but others have 1 (omniglot)?
I've seen others do:
transform = transforms.Compose([
lambda x: x.convert("RGB"),
transforms.Resize(224),
transforms.ToTensor(),
])
in pytorch. Is that what you do?