Solved by tom in post #8 
              
              
                Sorry for diverting your thread. 
Yes, the pruning seems to rename weights (appending _orig probably is the most common), you can get the current set of weight names using 
print([n for n, p in module.named_parameters()])

There is an equivalent named_buffers for non-parameters that gets masks. 
Als…