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 your custom layers.