DeepRec icon indicating copy to clipboard operation
DeepRec copied to clipboard

Can not get the value of embedding variables using NewCheckpointReader

Open Lihengwannafly opened this issue 3 years ago • 0 comments

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.

Lihengwannafly avatar Feb 21 '22 02:02 Lihengwannafly