Solved by ptrblck in post #4 
              
              
                Well, there is a way, but it’s a bit ugly, so sorry for that, but I can’t come up with a cleaner way at the moment. 
I’ve created a small example how to change the keys in the state_dict: 
# Create Sequential model
modelA = nn.Sequential(
    nn.Conv2d(3, 6, 3, 1, 1),
    nn.ReLU(),
    nn.Conv2d(6,…