I assume you’ve tried to create the new model by wrapping the child modules into an nn.Sequential container.
If that’s the case, note that you will lose all functional API calls from the forward method in your original model, e.g. out = out.reshape(out.size(0),-1) as well as the F.relu calls.
Thus you should add them via e.g. nn.Flatten, nn.ReLU.
Alternatively, you could manipulate the forward method or use forward hooks to get the desired activation.