Solved by ptrblck in post #4 
              
              
                I had something like this in mind: 
# save
model1 = nn.Linear(10, 5)
model2 = nn.Linear(5, 2)

model1 = torch.jit.script(model1)
model2 = torch.jit.script(model2)

torch.jit.save(model1, 'model1.pt')
torch.jit.save(model2, 'model2.pt')


# load
class MyModel(nn.Module):
    def __init__(self, model1…