From 35a7c163bc6928eae9c54f2a3e686ec308dcf03a Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Sun, 12 Jan 2020 03:12:19 +0100 Subject: tf utils to read vectors --- cli/app/utils/tf_utils.py | 10 ++++++++++ 1 file changed, 10 insertions(+) create mode 100644 cli/app/utils/tf_utils.py (limited to 'cli/app/utils') diff --git a/cli/app/utils/tf_utils.py b/cli/app/utils/tf_utils.py new file mode 100644 index 0000000..69a38e4 --- /dev/null +++ b/cli/app/utils/tf_utils.py @@ -0,0 +1,10 @@ +from tensorflow.python.training import py_checkpoint_reader + +def read_checkpoint(fp_ckpt, key): + reader = py_checkpoint_reader.NewCheckpointReader(fp_ckpt) + var_to_shape_map = reader.get_variable_to_shape_map() + var_to_dtype_map = reader.get_variable_to_dtype_map() + # for key, value in sorted(var_to_shape_map.items()): + # print("tensor: %s (%s) %s" % (key, var_to_dtype_map[key].name, value)) + # print(reader.get_tensor(key)) + return reader.get_tensor(key) -- cgit v1.2.3-70-g09d2