Hi, I cannot use torch vision because I design a own model. I just want to transfer my trained weight from binary classifier to other task likes four class problem. I just change number from 2 to 4 in the new network Net()
self.conv_classify= nn.Conv2d(50, 2, 1, 1, bias=True) 

to
self.conv_classify= nn.Conv2d(50, 4, 1, 1, bias=True) 

Other layers are copied weight from 10000_model.pth