Solved by tom in post #3 
              
              
                Can you be a bit more specific how your goal differs from what the fx show case for torchvision does? 
The snippet I have in mind is: 
from torchvision.models.feature_extraction import create_feature_extractor
# Confused about the node specification here?
# We are allowed to provide truncated node n…