Solved by ptrblck in post #2 
              
              
                If you try to create a new nn.Sequential model using the layers from Net, the flatten operation will be missing. 
Additionally, you are currently using only one pooling layer, as both are defined as self.pool. 
Probably the most straightforward way would be to derive another class from Net and just …