Solved by ptrblck in post #2 
              
              
                That’s great debugging! 
The reason for this is that you’ve forgotten to return out in the forward method of PreActResnetBlock. Adding it should solve the issue.