But if, for instance, I have a neural network with a structure like this:
layers1 = nn.sequential(…)
layers2 = nn.sequential(…)
layers3 = nn.sequential(…)
And I would like to save the model so that device1 loads layers1, device2 layers2…