Hello
I’m trying to remove the classification layer for the torchvision model resnet101-deeplabv3 for semantic seg but I’m having trouble getting this to work. I’ve tried using
backbone = nn.Sequential(*list(self.resnet101deeplab.children())[:-1])
In various ways with no luck. Part of the issue is it returns an OrderedDict and I’m unsure the proper way to take a layer away or extract the required ones. Can anyone help?