DeepRec
DeepRec copied to clipboard
Can not get the value of embedding variables using NewCheckpointReader
from tensorflow.python import pywrap_tensorflow
reader = pywrap_tensorflow.NewCheckpointReader(latest_checkpoint)
var_to_shape_map = reader.get_variable_to_shape_map()
for key in var_to_shape_map:
print(reader.get_tensor(key))
I want export the value of embedding variables, and I test it in nvtf successfully. But in deeprec, the value is [], an empty list.