Altering ResNet18 for single channel images

ptrblck November 9, 2018, 8:10pm 12

No worries. I had the same mistake quite a few times.
Good it’s working now!

Extracting feature vector for grey images via ResNet18: output with shape [1, 224, 224] doesn't match the broadcast shape [3, 224, 224]