Модификации вvalu.py для других архитектур в поэзах тензорного потока

Question

Модификации вvalu.py для других архитектур в поэзах тензорного потока

1

Там есть сценарий для измерения производительности модели при тензорном потоке-для-поэтов-2/scripts/evaluation.py. Этот сценарий был первоначально написан для моделей mobilenet, но может быть соответствующим образом изменен для других архитектур. Что я должен написать в строке 41

logits = graph.get_tensor_by_name("final_training_ops/Wx_plus_b/add:0").

Я получаю "KeyError:

Название final_training_ops/Wx_plus_b/add: 0 'относится к тензору, который не существует. Операция "final_training_ops/Wx_plus_b/add" не существует на графике.

duplex143 19 июнь 2018, в 08:45

Источник

Теги:

python

tensorflow

convolutional-neural-network

1 ответ

Ещё вопросы

duplex143 · Accepted Answer · 2018-07-02T04-49-00.000Z

Я понял это сам. Учитывая файл.pb, мы можем просматривать операции в замороженных графиках.pb TensorFlow, используя этот код. Итак, нам нужно заменить строку 41 последней операцией в соответствии с нашим.pb файлом.