hello again,
Can you please help me with how to use pretrained weights for different model. I have built a simple fcn model for segmentation with 5 blocks of conv, relu and pooling. I want to use pretrained weights of vgg/resent. I loaded the resnet pretrained weights as below. Is it the right way? I got better results, but I am not sure how the pretrained weights get added to my new model.
model = fcn()
model.load_state_dict(model_zoo.load_url(model_urls[‘resnet50’]), strict=False)
Is it based on layer names or automatically concat the weights with corresponding filter sizes. Please give me some inputs, since I am new to using pretrained weights.
Thank you