Both roi_head have same key names, it’s just the weights that are different. Even the shapes of in between layers is same. I just want to load the weights of roi_head with that of another model.
I want to just replace the weights of these :
'roi_heads.box_head.fc6.weight' 'roi_heads.box_head.fc6.bias' with that of another identical roi_head which will be used for taking predictions.