Hello everyone,
I’m trying to train images with one channel (That what I want for specific reason), but PyTorch refuses to deal with that. Is there any way to deal with one channel images? ’
This is the error that I’ve got.
RuntimeError: output with shape [1, 224, 224] doesn't match the broadcast shape [3, 224, 224]