Hi! I’m trying to implement the baseline model described in the Stanford MURA paper. The following is the code for my model:
model = densenet169(pretrained=True)
model.classifier = nn.Sequential(
nn.Linear(in_features=1664, out_features=1),
nn.Sigmoid()
)
model = model.cuda()
This works fine with the default input size of 224x224. However, the paper describes the input size to be 320x320. However, when input of this shape is fed to the model defined in the code above, it unsurprisingly throws a shape mismatch error. I’m new to PyTorch (and deep learning in general) and I’m having trouble implementing this. Can someone please help? I’ve looked at other solutions proposed for ResNet and other networks but I haven’t been able to apply any of those to my DenseNet169.