Is it possible to check each ouput of a layer in a nn.Sequential or nn.Module?

[maplewizard]

I want to check the output value of each layer I build. but I didn’t find any element in layers. Is there any way to do this?

For example, I have a model:

model = nn.Sequential(nn.Linear(128, 64), nn.Linear(64, 32), nn.Linear(32, 1))
av = Variable(torch.rand(7, 128), requires_grad = True)
model.forward(av)

I want to see the output of each linear layer.

[alexis-jacq (Alexis David Jacq)]

You should have a look here:

How to extract features of an image from a trained model

Hi all, I try examples/imagenet of pytorch. It is awesome and easy to train, but I wonder how can I forward an image and get the feature extraction result? After I train with examples/imagenet/main.py, I get model as, model_best.pth.tar And I load this file with model = torch.load('model_best.pth.tar') which gives me a dict. How can I use forward method to get a feature (like fc7 layer’s output of Alexnet) of an image with this model?