You could register a forward hook on model.fc3 as described here or alternatively manipulate the forward method and return the output activation of self.fc3 as well as from self.out.
13 Likes
You could register a forward hook on model.fc3 as described here or alternatively manipulate the forward method and return the output activation of self.fc3 as well as from self.out.