Hello,
Given the following code:
def forward(self, x):
x = x.transpose(2,1)
x = self.max_pool(x)
x = x.view(x.size(0), -1)
x = self.classifier(x)
Does exist an elegant way to get back the results of each layer? Or I should use a dictionary containing a key for each layer result?
Thank you