Не удается найти переменную, которая вводится в ReadVariableOp

Question

Попытка сохранить файл Keras .h5, содержащий веса, в файл Tensorflow .pb

# I keep getting the error:  ValueError: Cannot find the variable that is an input to the ReadVariableOp.
frozen_graph = freeze_session(K.get_session(),
                              output_names=[out.op.name for out in model.keras_model.output])

0

frozen_graph = freeze_session (K.get_session (), output_names = [out.op.name для out в model.keras_model.output])

Я получил ошибку:

ValueError Traceback (последний последний вызов) в 1 frozen_graph = freeze_session (K.get_session (), ----> 2 output_names = [out.op.name для out в model.keras_model.output])

в freeze_session (сеанс, keep_var_names, output_names, clear_devices) 26 node.device = ""27 frozen_graph = tf.graph_util.convert_variables_to_constants (---> 28 сессий, input_graph_def, output_names, freeze_var_names) 29 return frozen_graph

~ / anaconda3 / envs / env_name / lib / python3.6 / site-packages /ensorflowflow/python/util/deprecation.py в new_func (* args, ** kwargs) 322 «в будущей версии», если дата «Нет» («после% s»% date), 323 инструкции) -> 324 return func (* args, ** kwargs) 325 return tf_decorator.make_decorator (326 удовольствияc, new_func, 'deprecated',

~ / anaconda3 / envs / env_name / lib / python3.6 / site-packages /ensorflow / python / framework / graph_util_impl.py в convert_variables_to_constants (sess, input_graph_def, output_node_namesvariable_names_whitelist, variable_names_blacklist) 300 source_op_name = get_input_name (map_name_to_node [source_op_name]) 301 if map_name_to_node [source_op_name] .op! = "VarHandleOp": -> 302 вызвать ValueError («не удается найти переменную», которая не может найти переменную («не может найти» переменную). ") 304

ValueError: Невозможно найти переменную, которая является входом для ReadVariableOp.

Answer 1

Я только что столкнулся с той же проблемой, добавив

import keras.backend as K
k.set_learning_phase(0)

, которая переводит фазу обучения в режим тестирования, и было решением.