Is it possible to check each ouput of a layer in a nn.Sequential or nn.Module?
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: <SCODE>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)
<ECODE> I want to see the output of each linear layer.
maplewizard
alexis-jacq