My model is resnet20 , there are parameters from conv2d 、fc and BN ,but I just want to load the conv2d weight parameters from the pretrained model , how can I sovle this problem?
Thanks for your help
My model is resnet20 , there are parameters from conv2d 、fc and BN ,but I just want to load the conv2d weight parameters from the pretrained model , how can I sovle this problem?
Thanks for your help