pytroch获取中间变量/feature

方法1

使用torchvision.models.feature_extraction的方法create_feature_extractor

from torchvision.models.feature_extraction import create_feature_extractor
import torch
import torchvision

model = torchvision.models.resnet18()

for n,m in model.named_modules()