Solved by ptrblck in post #2 
              
              
                You could apply all child modules, if the model would generally work in an nn.Sequential container. 
The model would thus have to define only nn.Modules in its __init__ method and apply these modules sequentially. 
However, this is often not the case and the forward method often uses functional call…