The model is nearly identical to the torchvision resnet code. The error is raised at the first convolution of the ResNet model, namely:
x = self.conv1(x)

which is defined as:
self.conv1 = nn.Conv2d(3, self.inplanes, kernel_size=7, stride=2, padding=3, bias=False)

The input shape is (1, 224, 224, 3).
I use a GTX Titan X, but also tested a GTX1060 with the same result.