Я пытаюсь получить вывод среднего уровня (сохраненный в глобальной переменной, которую я определил), используя функцию ловушки в Pytorch.Однако после выполнения модели NN переменная, которую я определил, пуста.
# features are used for storing the middle layer output
features = []
# hook function,
# output is a tuple, so I change it into a list
def hook (module, input, output):
features = deepcopy(list(output))
handle = self.model.roi_heads.box.feature_extractor.pooler[3]
with torch.no_grad():
predictions = self.model(image_list)
print(features)
handle.remove()
Ожидается получение выходных данных этого среднего уровня, однако функции пусты.