Solved by ptrblck in post #4 
              
              
                I would create the custom module, load the resnet inside and write the forward method as you wish. 
Once this module works as you want it to, save the state_dict and reload this one. 
This approach will make it easier to use your custom module without having to map the original resnet parameters to …