Could you explain your use case a bit more?
Are you using a custom module, which contains 3 separate resnets?
If so, you should be able to access the desired submodel directly e.g. via: model.resnet1.state_dict().