Solved by ptrblck in post #2 
              
              
                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 gettin…