I am looking to do some experiments in transfer learning using Resnet18. Currently I replace the fully connected layer as follows :
fc_inputs = model.fc.in_features
model.fc = nn.Sequential(nn.Linear(fc_inputs, num_classes),)
I am trying to figure out how to remove conv5_x and connect the fully connected layer to the conv4_x layer. ( Or connect it to conv3_x )
I want to compare the results tapping into the network at different places.

794×717