Solved by ptrblck in post #2 
              
              
                I would not recommend to wrap random models into an nn.Sequential container, since you would lose all operations from the functional API in the forward method. 
While your approach might work for the resnet, the densenet would miss these ops. 
The better way would be to either derive your custom mod…