I want to train RGB images on ResNet50 using transfer learning method.
I was resized images to 224x224 and I got error using GPU:
RuntimeError: output with shape [1, 224, 224] doesn't match the broadcast shape [3, 224, 224]
On CPU all works fine.
How to deal with it?