2016-10-06 7 views
3

Что такое простой способ доступа к внутренним нейронам нейронной сети в факеле?Доступ к внутренним нейронам факела Нейронная сеть

Я обучил многослойную нейронную сеть, используя nn библиотеку torch7. Я хочу посмотреть, каковы выходные значения некоторых конкретных внутренних нейронов, когда какой-либо вход пересылается в сеть, или изменять результаты вручную.

Возможно ли это? Если это так, это простые решения для сети, которая уже обучена и сохранена? Причина в том, что моя нейронная сеть является автокодировщиком, поэтому нейроны в середине важны - вид вывода всей сети.

ответ

2

если ваш нейронная сеть называется Nnet и слой интерес middle_module, то сделать это:

middle_module = nn.Linear(...) 
nnet = nn.Sequential(...) 
nnet:forward(your_data) 
local hidden_output = middle_module.output 

Переменная output экземпляр модуля яя всегда сохраняется. Это связано с деталями реализации backpropagation.

Редактировать: Я понимаю, что вы спрашивали о уже сохраненных моделях. Вы можете использовать nn.Sequential.get, если знаете индекс интересующего модуля.

Смежные вопросы