Wrapping the child modules into nn.Sequential containers might not work for a lot of modules, as they might use functional calls in their original forward method, which would be lost.
You could derive a new model and manipulate the forward method instead (unsure, what kind of errors you were getting since returning the intermediates shouldn’t raise a shape mismatch) or you could use e.g. forward hooks alternatively.