Nilakshan_Kunanantha:

elevant layer and donthe infe


To get intermediate outputs you can use hooks.
Here is a tutorial

  
      
      Medium – 12 May 20
  
  
    

The One PyTorch Trick Which You Should Know 8

How hooks can improve your workflow significantly

  Reading time: 4 min read