Hi everyone
I have a kind of funny question
Who knows that how can I save (store) output of a layer ?
Codes below represents my NN. Basically I would like to save last layer of my encoder to a tensor
#Autoencoder and Autodecoder conv layer…
class Autoencoder(nn.Module):
def init(self):
super(Autoencoder,self).init()
self.encoder = nn.Sequential(
nn.Conv1d(1,3,kernel_size=4, stride=2),
nn.MaxPool1d(3, stride=2),
nn.Tanh(),
nn.Conv1d(3,6,kernel_size=4, stride = 2),
nn.MaxPool1d(3,stride=1),
nn.Tanh(),
nn.Conv1d(6,9,kernel_size=3, stride=1),
nn.MaxPool1d(2,stride = 1),
nn.Tanh(),
nn.Conv1d(9,12,kernel_size = 3, stride = 1),
nn.MaxPool1d(2,stride = 1),
nn.Tanh())
self.decoder = nn.Sequential(
nn.ConvTranspose1d(12,9,kernel_size = 3, stride = 2),
nn.Tanh(),
nn.ConvTranspose1d(9,6,kernel_size = 4,stride = 1),
nn.Tanh(),
nn.ConvTranspose1d(6,3,kernel_size= 8,stride = 2),
nn.Tanh(),
nn.ConvTranspose1d(3,1,kernel_size = 8, stride = 4),
nn.Tanh())
def forward(self,x):
x = self.encoder(x)
x = self.decoder(x)
return x
Thanks all