Как извлечь веса и другие значения переменных из контрольной точки tenorflow без восстановления графика?

Question

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

Итак, не восстанавливая график, как извлечьзначения хранятся в контрольной точке.Я мог бы потенциально преобразовать все из контрольной точки в словарь из бесчисленных массивов или что-то подобное.

Answer 1

Нашел решение:

reader = tf.train.NewCheckpointReader("/path/to/checkpoint")
shapes_dict = reader.get_variable_to_shape_map()  # use it to get the variable names
extracted_values = reader.get_tensor(shapes_dict.keys()[0])
# array([[ 0.       , -1.8053141],
#        [-1.5647348,  0.       ]], dtype=float32)

tf.train.NewCheckpointReader не совсем задокументировано в текущей документации API r1.12.Но вы можете увидеть пример использования в исходном коде здесь .