Solved by KFrank in post #2 
              
              
                Hello Al! 

When you create new_model it doesn’t know about the detailed 
structure of your class Net.  In particular, it doesn’t know about 
Net's forward() function.  (It does know about model.children()). 
So I don’t think any of the torch.nn.functional functions used 
in model (an instance of yo…