My input is now [batch_size, 1, 244, 244] as my images are only single channel.
I’m not sure how to output the stack trace but I can give it a try!
My model manipulation is as follows:
resnet18 = models.resnet18()
resnet18.conv1 = torch.nn.Conv2d(1, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
resnet18.fc = = torch.nn.Linear(512, 1)

As suggested by David below I’ll output the input sizes just to be sure though.