Solved by ptrblck in post #2 
              
              
                You can just use a plot library like matplotlib to visualize the output. 


Sure! You could use some loss function like nn.BCELoss as your criterion to reconstruct the images. 


Forward hooks are a good choice to get the activation map for a certain input. 


Here is a small code example as a sta…