def get_activation(self, layer_index):
    return self.model.layer_output(self.net_input, layer_index)
def get_activation(self, layer_index):
    return self.model.layer_output(self.net_input, layer_index)